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.