Apply different Transform (Data Augmentation) to Train and Validation

[xinqi (Frank)]

Thanks.

Should I still use ImageFolder to obtain the variable train_data?
For example train_data = datasets.ImageFolder(base_path + '/train/') without transfrom