ptrblck:

mation only once.
I’m not sure, what your custom dataset is supposed to do besides applying another transformation, which will create this mismatch.


Okay I am trying and let you know. I think I need to pass two tranormations: first transformation (to Tensor and Normalisation for data only ) and last transformation (data augmentation) as it is.