Transforms on subset

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

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

Thanks. It works like a charm.