How does WeightedRandomSampler work?

[cuixing158_1 (cuixing)]

your understanding may be correct,pls reference here ,perhps it helps.

Balanced Sampling between classes with torchvision DataLoader

you should look at How to Prevent Overfitting