Minor fixes for #47

2024-03-25 18:50:47 +01:00
parent f00252552a
commit c5635b7d94
37 changed files with 13 additions and 18 deletions
--- a/lerobot/scripts/train.py
+++ b/lerobot/scripts/train.py
@@ -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]