Is there a way to make the DataLoader produce batches containing only one class each?
For example, the training dataset could contain 1000 images A, 1500 images B, 1300 images C…
I would need the DataLoader to then yield a batch of only A samples, then another batch of B samples, etc.
Thank you so much for any help or tips!