fix simxarm factory

2024-02-22 13:04:24 +00:00
parent 96c53ad06f
commit 63d18475cc
3 changed files with 1 additions and 4 deletions
--- a/lerobot/scripts/train.py
+++ b/lerobot/scripts/train.py
@@ -80,7 +80,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
            alpha=cfg.per_alpha,
            beta=cfg.per_beta,
            num_slices=num_traj_per_batch,
-            strict_length=False,
+            strict_length=True,
        )

        online_buffer = TensorDictReplayBuffer(