Balanced Sampling between classes with torchvision DataLoader

you should look at How to Prevent Overfitting