Question about Transforms

I have a dataset with 3 classes. To train a model, I want to apply different transforms for the 3 classes. How do I do that?

You can create a custom Dataset as described here and apply custom transformations in the __getitem__ by using a condition on the target.