StefanCepa995:

problem is in the transform component because it tries to normalize images that are initially grayscale, and not RGB:


I’m facing the same issue, have you found a workaround for this?