Thanks, it works, I think that the I was using for extracting single batch was taking single batch from whole data and not restricted to the subset sampler, when I tried :for data, target in loader: it is the same each epoch