If your data are images try with the Image Folder class, if not I’m sorry I cant help. My data is all inclusive in a csv so I read the csv and then I feed the dataset class with the data and the labels so I get the data prepared for the dataloader