WeightedRandomSampler has low data loading efficiency

[ptrblck]

This topic might be helpful asking about a weighted and distributed sampler.