Increase number of images

One-hot encoded targets are not expected and you should transform them to class indices via target = torch.argmax(target, 1). An example of using WeightedRandomSampler can be found here.