Update experiment config files to Evaluator

alexhernandezgarcia · Jun 11, 2024 · 183fd4a · 183fd4a
1 parent b4c3467
commit 183fd4a
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 24 deletions.
diff --git a/config/experiments/ccube/branin.yaml b/config/experiments/ccube/branin.yaml
@@ -51,6 +51,13 @@ policy:
     shared_weights: True
     checkpoint: backward
 
+# Evaluator
+evaluator:
+  first_it: False
+  period: 500
+  n: 1000
+  checkpoints_period: 500
+
 # WandB
 logger:
   do:
@@ -62,13 +69,8 @@ logger:
     - continuous
     - ccube
     - branin
-  test:
-    period: 500
-    n: 1000
-  checkpoints:
-    period: 500
 
 # Hydra
 hydra:
   run:
-    dir: ${user.logdir.root}/ccube/branin/${now:%Y-%m-%d_%H-%M-%S}
+    dir: ${user.logdir.root}/ccube/branin/${now:%Y-%m-%d_%H-%M-%S_%f}
diff --git a/config/experiments/ccube/hartmann.yaml b/config/experiments/ccube/hartmann.yaml
@@ -54,6 +54,13 @@ policy:
     n_layers: 3
     checkpoint: backward
 
+# Evaluator
+evaluator:
+  first_it: False
+  period: 500
+  n: 1000
+  checkpoints_period: 500
+
 # WandB
 logger:
   do:
@@ -65,13 +72,8 @@ logger:
     - continuous
     - ccube
     - hartmann
-  test:
-    period: 500
-    n: 1000
-  checkpoints:
-    period: 500
 
 # Hydra
 hydra:
   run:
-    dir: ${user.logdir.root}/ccube/hartmann/${now:%Y-%m-%d_%H-%M-%S}
+    dir: ${user.logdir.root}/ccube/hartmann/${now:%Y-%m-%d_%H-%M-%S_%f}
diff --git a/config/experiments/grid/branin.yaml b/config/experiments/grid/branin.yaml
@@ -47,6 +47,13 @@ policy:
     shared_weights: True
     checkpoint: backward
 
+# Evaluator
+evaluator:
+  first_it: False
+  period: 500
+  n: 1000
+  checkpoints_period: 500
+
 # WandB
 logger:
   do:
@@ -57,13 +64,8 @@ logger:
     - gflownet
     - grid
     - branin
-  test:
-    period: 500
-    n: 1000
-  checkpoints:
-    period: 500
 
 # Hydra
 hydra:
   run:
-    dir: ${user.logdir.root}/grid/branin/${now:%Y-%m-%d_%H-%M-%S}
+    dir: ${user.logdir.root}/grid/branin/${now:%Y-%m-%d_%H-%M-%S_%f}
diff --git a/config/experiments/grid/hartmann.yaml b/config/experiments/grid/hartmann.yaml
@@ -50,6 +50,13 @@ policy:
     n_layers: 3
     checkpoint: backward
 
+# Evaluator
+evaluator:
+  first_it: False
+  period: 500
+  n: 1000
+  checkpoints_period: 500
+
 # WandB
 logger:
   do:
@@ -60,13 +67,8 @@ logger:
     - gflownet
     - grid
     - hartmann
-  test:
-    period: 500
-    n: 1000
-  checkpoints:
-    period: 500
 
 # Hydra
 hydra:
   run:
-    dir: ${user.logdir.root}/grid/hartmann/${now:%Y-%m-%d_%H-%M-%S}
+    dir: ${user.logdir.root}/grid/hartmann/${now:%Y-%m-%d_%H-%M-%S_%f}