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

backup wip

This commit is contained in:
Alexander Soare 2024-05-17 16:52:59 +01:00
parent 918868162e
commit 2169a8d146
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/scripts/train.py
View File

@ -444,6 +444,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
policy, policy,
n_episodes=1, n_episodes=1,
return_episode_data=True, return_episode_data=True,
# TODO(now): Actually we shouldn't be seeding this every time!
start_seed=cfg.training.online_env_seed, start_seed=cfg.training.online_env_seed,
enable_progbar=False, enable_progbar=False,
) )