Solved by ptrblck in post #2 
              
              
                For the SubsetRandomSampler you would need to provide indices, so we have to get the balanced indices from WeightedRandomSampler. 
One way would be to create the sampler and instead of returning the data we could return the indices and store them somehow. This doesn’t really sound like a good approa…