Update ccube config files because policy config is not being loaded a…

…s expected
alexhernandezgarcia · Nov 16, 2023 · 7413040 · 7413040
1 parent 01a1c08
commit 7413040
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 24 deletions.
diff --git a/config/experiments/ccube/corners.yaml b/config/experiments/ccube/corners.yaml
@@ -40,18 +40,17 @@ gflownet:
     z_dim: 16
     lr_z_mult: 100
     n_train_steps: 10000
-  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: 128
+    n_layers: 2
+    checkpoint: forward
+  backward:
+    shared_weights: True
+    checkpoint: backward
 
 # WandB
 logger:

diff --git a/config/experiments/ccube/uniform.yaml b/config/experiments/ccube/uniform.yaml
@@ -40,18 +40,17 @@ gflownet:
     z_dim: 16
     lr_z_mult: 100
     n_train_steps: 10000
-  policy:
-    forward:
-      type: mlp
-      n_hid: 256
-      n_layers: 3
-      checkpoint: forward
-    backward:
-      type: mlp
-      n_hid: 256
-      n_layers: 3
-      shared_weights: False
-      checkpoint: backward
+
+# Policy
+policy:
+  forward:
+    type: mlp
+    n_hid: 128
+    n_layers: 2
+    checkpoint: forward
+  backward:
+    shared_weights: True
+    checkpoint: backward
 
 # WandB
 logger: