Solved by JuanFMontesinos in post #4 
              
              
                If I properly understood, you replaced CIFAR10 dataset by your own by loading them from CIFAR10 dataset right? 
The point is that you are getting the data somehow from the dataset directory rather than iterating the dataset. My guess is that CIFAR10 call function is never run, consequently, transfor…