drop_last=False

2024-04-10 14:59:54 +00:00 · 2024-04-10 14:59:54 +00:00 · 693f620df0
parent 79542ecd13
commit 693f620df0
1 changed files with 2 additions and 2 deletions
--- a/lerobot/scripts/train.py
+++ b/lerobot/scripts/train.py
@ -236,7 +236,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
        batch_size=cfg.policy.batch_size,
        shuffle=True,
        pin_memory=cfg.device != "cpu",
-        drop_last=True,
+        drop_last=False,
    )
    dl_iter = cycle(dataloader)

@ -283,7 +283,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
        batch_size=cfg.policy.batch_size,
        sampler=sampler,
        pin_memory=cfg.device != "cpu",
-        drop_last=True,
+        drop_last=False,
    )
    dl_iter = cycle(dataloader)