I am having a dataset with multiple classes with each class having multiple images.
While running the loop over data loader I want to get two images from each class in every iteration.
I would like to have an iterator like
for image1, image2, label in dataloader:
Please let me know how can I achieve this.