Balanced Sampling between classes with torchvision DataLoader

[ptrblck]

Double post from here. Let’s stick to the newly created topic for further discussion.