wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update lerobot/scripts/train.py 

Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
This commit is contained in:
Remi 2024-05-30 12:03:25 +02:00 committed by GitHub
parent d6abce38eb
commit 752147c5af
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/scripts/train.py
View File

@ -422,7 +422,8 @@ def train(cfg: DictConfig, out_dir: str | None = None, job_name: str | None = No
logging.info("End of online training") logging.info("End of online training")
eval_env.close() if cfg.training.eval_freq > 0:
eval_env.close()
online_training_env.close() online_training_env.close()