Jordan_Howell:

I’m not sure how batches works with the custom datasets and how to get them to pull correctly.


The Dataset.__getitem__ will get an index as the argument from the DataLoader to create a batch.
These indices will be in the range [0, len(dataset)-1], so you would only have to make sure to load each corresponding sample using the passed index.
I would still recommend to check the path and make sure that the idx is used to load the right path.