thank you. I’d have to look at the source with this new info. However, there is a batch_size argument for WeightedRandomSampler(), I wonder why you need num_samples as well? are there situations where you’d use both?? combined with the fact that it is passed to a DataLoader() which has its own batch_size… I’ll just say this is a confusing interface.