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

fix online training

This commit is contained in:
Cadene 2024-04-19 23:48:43 +00:00
parent 85f1554da8
commit b2b5329683
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/train.py
View File

@ -344,7 +344,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
sampler, sampler,
hf_dataset=eval_info["episodes"]["hf_dataset"], hf_dataset=eval_info["episodes"]["hf_dataset"],
episode_data_index=eval_info["episodes"]["episode_data_index"], episode_data_index=eval_info["episodes"]["episode_data_index"],
online_pc_sampling=cfg.get("demo_schedule", 0.5), pc_online_samples=cfg.get("demo_schedule", 0.5),
) )
for _ in range(cfg.policy.utd): for _ in range(cfg.policy.utd):