From 183fd4acea7834d09962da77c6d31ae92534d566 Mon Sep 17 00:00:00 2001
From: alexhernandezgarcia <alexhdezgarcia@gmail.com>
Date: Tue, 11 Jun 2024 12:45:13 -0400
Subject: [PATCH] Update experiment config files to Evaluator

---
 config/experiments/ccube/branin.yaml   | 14 ++++++++------
 config/experiments/ccube/hartmann.yaml | 14 ++++++++------
 config/experiments/grid/branin.yaml    | 14 ++++++++------
 config/experiments/grid/hartmann.yaml  | 14 ++++++++------
 4 files changed, 32 insertions(+), 24 deletions(-)

diff --git a/config/experiments/ccube/branin.yaml b/config/experiments/ccube/branin.yaml
index b490b8ee8..377027ed3 100644
--- 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
index a3cb1cd92..192f279f2 100644
--- 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
index 6245b872c..e9d546aa9 100644
--- 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
index a130154ba..f395e1b9e 100644
--- 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}