Dataset/Dataloader minibatch class distribution

Answered here.