Solved by ptrblck in post #4 
              
              
                If you don’t want to transform your images, you could load them as tensors. 
However, if you do need a transformation, you would have to transform the tensors back to PIL.Images, apply the augmentation/transformation, and transform them back to tensors. 
This shouldn’t be too bad regarding the perfo…