Solved by ptrblck in post #2 
              
              
                I assume scaling is done for normalization. If so, you should split your data first and then scale it. Otherwise you will leak the test information into your training procedure. 


Have a look at the Data loading tutorial. Basically you create a Dataset and add your data augmentation in the __geti…