@ptrblck I have a question:
with plain sampler I needed 10GB gpu memory for the batch size of 64
However, with WeightedRandomSampler same batch size is using only 2.3GB gpu memory.
does it has to do with repetitions? In fact, my data is heavily imbalanced.

Or should I be concerned?
Thanks!