DataLoader: direct multi-index instead of coallate for batches on map-style datasets

[ptrblck]

You could try to use a BatchSampler as given in this code snippet, which would pass multiple indices to the __getitem__ and thus you could load multiple samples in this call.