Apply different Transform (Data Augmentation) to Train and Validation

[gregunz]

You’re welcome

Should I still use ImageFolder to obtain the variable train_data ?

Yes, exactly!

If this solves your issue please mark it as solved with the previous answer as solution