Transforms on subset

data
1

I created a test_train split using torch.utils.data.Subset. How do I apply different train/test transforms on these before passing them as an argument to the Dataloader?

2

One approach would be to write a simple custom Dataset, which just applies your transformation to the internal subset as described here.

3

Thanks. It works like a charm.