manurare:

So if input belongs to C1 the classifier should be used to see T1-like images and not T2-like as in training never happened a C1 input to be fed into N2. But actually it seems that my classifier did not learn about class features of the image but only about the transformation. I don’t know if my thoughts are correct.


That’s what I’m concerned about.
You are using the targets to switch between different transformations as well as different model paths, which would not be possible in your validation and test use cases.