Luca_Pamparana:

train_folder_dataset = dset.ImageFolder(root=self.data_path)



I’m doing something similar, but without the dset.ImageFolder class (because I don’t have images).
What I did was preprocess into:
data/train/
data/val/
data/test/

at the beginning of my pipeline, so all nets would have the same view on what is train/val/test