Thanks for the quick reply.
Ok, I will try that, but I am a bit confused by the documentation-how would I modify my dataloader below to add in the weightedRandomSampler?
testloader = data_utils.DataLoader(test_dataset, batch_size=20, shuffle=True)
Could you provide some guidance on usage, maybe with an example?
Thanks again