diff --git a/lerobot/scripts/train.py b/lerobot/scripts/train.py index 9bf49c05..eb33b268 100644 --- a/lerobot/scripts/train.py +++ b/lerobot/scripts/train.py @@ -422,7 +422,8 @@ def train(cfg: DictConfig, out_dir: str | None = None, job_name: str | None = No logging.info("End of online training") - eval_env.close() + if cfg.training.eval_freq > 0: + eval_env.close() online_training_env.close()