Single agent total iteration limit

utiasDSL · Feb 27, 2021 · 68a9612 · 68a9612
1 parent 191443b
commit 68a9612
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/experiments/learning/singleagent.py b/experiments/learning/singleagent.py
@@ -236,13 +236,13 @@
                                  verbose=1,
                                  best_model_save_path=filename+'/',
                                  log_path=filename+'/',
-                                 eval_freq=int(5000/ARGS.cpu),
+                                 eval_freq=int(2000/ARGS.cpu),
                                  deterministic=True,
                                  render=False
                                  )
-    model.learn(total_timesteps=int(1e12),
+    model.learn(total_timesteps=35000, #int(1e12),
                 callback=eval_callback,
-                log_interval=100
+                log_interval=100,
                 )
 
     ### Save the model #########################################