search

MarkFzp / act-plus-plus

Imitation learning algorithms with Co-training for Mobile ALOHA: ACT, Diffusion Policy, VINN
https://mobile-aloha.github.io/
MIT License
2.86k stars 525 forks source link

shuffle dataloader? #50

Closed kayburns closed 2 months ago

kayburns commented 2 months ago

https://github.com/MarkFzp/act-plus-plus/blob/26bab0789d05b7496bacef04f5c6b2541a4403b5/utils.py#L268C64-L268C83

By default the torch DataLoader class does not shuffle: https://pytorch.org/docs/stable/data.html

Should DataLoader be modified to take shuffle=True?