Hey, I do have the same question, I define my train dataset first, which have transformations, then split it into train, val, test dataset.
I did try
test_dataset.transform = new_transforms

But, it did not work