Update discrete-matbench config files because policy config is not be…

…ing loaded as expected
alexhernandezgarcia · Nov 16, 2023 · 619281d · 619281d
1 parent 7ef6764
commit 619281d
Showing 1 changed file with 14 additions and 12 deletions.
diff --git a/config/experiments/workshop23/discrete-matbench.yaml b/config/experiments/workshop23/discrete-matbench.yaml
@@ -36,18 +36,20 @@ gflownet:
     n_train_steps: 10000
     lr_decay_period: 1000000
   replay_sampling: weighted
-  policy:
-    forward:
-      type: mlp
-      n_hid: 512
-      n_layers: 5
-      checkpoint: forward
-    backward:
-      type: mlp
-      n_hid: 512
-      n_layers: 5
-      shared_weights: False
-      checkpoint: backward
+
+# Policy
+policy:
+  forward:
+    type: mlp
+    n_hid: 512
+    n_layers: 5
+    checkpoint: forward
+  backward:
+    type: mlp
+    n_hid: 512
+    n_layers: 5
+    shared_weights: False
+    checkpoint: backward
 
 # WandB
 logger: