hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Minor fixes for #47

Browse Source
This commit is contained in:
Simon Alibert
2024-03-25 18:50:47 +01:00
parent f00252552a
commit c5635b7d94
37 changed files with 13 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/train.py
View File

@@ -231,7 +231,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
num_parallel_env = rollout.batch_size[0]
if num_parallel_env != 1:
# TODO(rcadene): when num_parallel_env > 1, episode needs to be incremented and we need to add tests
# TODO(rcadene): when num_parallel_env > 1, rollout["episode"] needs to be properly set and we need to add tests
raise NotImplementedError()
num_max_steps = rollout.batch_size[1]