preprocess function for validation is used for train dataset as well

[AlirezAkbary]

Hi. It looked like on this line, the preprocess_train should be passed to the get_dataset_fn instead of preprocess_val, as far as I understood.

CLIP-MIA/data.py

Line 484 in 1d8317a

args, preprocess_val, is_train=True, epoch=epoch)