feat: update

agent-dreamer_v3.py

@@ -26,6 +26,8 @@ def main(cfg_path: str, checkpoint_path: str, test=False):
     # Override configs for evaluation
     # You do not need to capture the video since you are submitting the agent and the video is recorded by DIAMBRA
     cfg.env.capture_video = False
+    # Only one environment is used for evaluation
+    cfg.env.num_envs = 1
 
     # Instantiate Fabric
     # You must use the same precision and plugins used for training.

agent-ppo.py

@@ -23,8 +23,9 @@ def main(cfg_path: str, checkpoint_path: str, test=False):
     print("Config parameters = ", json.dumps(cfg, sort_keys=True, indent=4))
 
     # Override configs for evaluation
-    if not test:
-        cfg.env.capture_video = False
+    # You do not need to capture the video since you are submitting the agent and the video is recorded by DIAMBRA
+    cfg.env.capture_video = False
+    # Only one environment is used for evaluation
     cfg.env.num_envs = 1
 
     # Instantiate Fabric