Solved by ptrblck in post #6 
              
              
                If you want to apply a new scaling for the complete batch you could either resample the data inside the training loop or use an approach, where the sampler might pass the index as well as the new scaling to the Dataset. 
I haven’t tested this approach so let me know, if you get stuck.