Load optimizer state in config

thatguy11325 · Apr 8, 2024 · a5e9648 · a5e9648
1 parent 19423f0
commit a5e9648
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/config.yaml b/config.yaml
@@ -26,6 +26,7 @@ debug:
     verbose: True 
     env_pool: False
     log_frequency: 5000
+    load_optimizer_state: False
 
 env:
   headless: True
@@ -43,7 +44,6 @@ env:
   reduce_res: True
   two_bit: True
   log_frequency: 2000
-  load_optimizer_state: False
 
 train:
   seed: 1
@@ -84,6 +84,7 @@ train:
   overlay_interval: 200
   cpu_offload: True
   pool_kernel: [0]
+  load_optimizer_state: False
 
 wrappers:
   baseline: