From 693f620df04e6a8523f88b323e1dff9881a1a672 Mon Sep 17 00:00:00 2001 From: Cadene Date: Wed, 10 Apr 2024 14:59:54 +0000 Subject: [PATCH] drop_last=False --- lerobot/scripts/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/scripts/train.py b/lerobot/scripts/train.py index 67e451d4..03506f2a 100644 --- 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)