wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

drop_last=False

This commit is contained in:
Cadene 2024-04-10 14:59:54 +00:00
parent 79542ecd13
commit 693f620df0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/train.py
View File

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