[Feature] Run evaluation on first iteration (#108)

* amend * amend
facebookresearch · Jul 9, 2024 · 2d7ac79 · 2d7ac79
1 parent c37938e
commit 2d7ac79
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.github/unittest/install_vmas.sh b/.github/unittest/install_vmas.sh
@@ -1,4 +1,4 @@
 
-python -m pip install vmas
+python -m pip install vmas matplotlib==3.8
 sudo apt-get update
 sudo apt-get install python3-opengl xvfb
diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py
@@ -657,7 +657,10 @@ def _collection_loop(self):
             # Evaluation
             if (
                 self.config.evaluation
-                and (self.total_frames % self.config.evaluation_interval == 0)
+                and (
+                    self.total_frames % self.config.evaluation_interval == 0
+                    or self.n_iters_performed == 0
+                )
                 and (len(self.config.loggers) or self.config.create_json)
             ):
                 self._evaluation_loop()