Solved by ptrblck in post #2 
              
              
                It might work if you manipulate the weights attribute, however I also think you shouldn’t see any performance regression if you create a new sampler and a new DataLoader after the epoch, as it should be relatively cheap compared to the iterations.