From 2dc3962bb8202a8dbc917703ff081ce5d5555e7b Mon Sep 17 00:00:00 2001 From: Matteo Bettini Date: Fri, 30 Aug 2024 10:19:38 +0200 Subject: [PATCH] amend --- benchmarl/experiment/experiment.py | 1 + 1 file changed, 1 insertion(+) diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py index 12fa8859..bee1900c 100644 --- a/benchmarl/experiment/experiment.py +++ b/benchmarl/experiment/experiment.py @@ -583,6 +583,7 @@ def _collection_loop(self): for _ in range( self.n_iters_performed, self.config.get_max_n_iters(self.on_policy) ): + iteration_start = time.time() if not self.config.collect_with_grad: batch = next(iterator)