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.
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.