Hi there! Have you tried creating a custom dataset to create the dataloader??
It’s a pytorch class, here are some examples:
https://pytorch.org/tutorials/beginner/data_loading_tutorial.html 14
https://stanford.edu/~shervine/blog/pytorch-how-to-generate-data-parallel 5
https://pytorch.org/docs/stable/data.html#torch.utils.data.Dataset 3
It was really useful to me!