From c2e725f309e3d724c908cd0240cca7845ca08012 Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Tue, 9 Jul 2024 14:54:47 +0100 Subject: [PATCH] amend --- benchmarl/experiment/experiment.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py index 1d8a03a2..c43cc8ca 100644 --- 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()