@Antonio_Ossa I filed Clarify the behavior of DataLoader sampler and batch_sampler parameters · Issue #71872 · pytorch/pytorch · GitHub 4 for the documentation issue. Feel free to comment there as needed.