use evaluator

alexhernandezgarcia · Mar 5, 2024 · dce8abf · dce8abf
1 parent 8139ab0
commit dce8abf
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/config/experiments/icml23/ctorus.yaml b/config/experiments/icml23/ctorus.yaml
@@ -2,6 +2,7 @@
 
 defaults:
    - override /env: ctorus
+   - override /evaluator: base
    - override /gflownet: trajectorybalance
    - override /proxy: torus
    - override /logger: wandb
@@ -33,6 +34,12 @@ gflownet:
     lr_z_mult: 1000
     n_train_steps: 5000
 
+# Evaluator
+evaluator:
+  period: 25
+  n: 1000
+  checkpoints_period: 500
+
 # Policy
 policy:
   forward:
@@ -51,15 +58,10 @@ policy:
 logger:
   lightweight: True
   project_name: "Continuous GFlowNet"
-  tags: 
+  tags:
     - gflownet
     - continuous
     - ctorus
-  test:
-    period: 25
-    n: 1000
-  checkpoints:
-    period: 500
 
 # Hydra
 hydra: