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)