Hi, it depends on the dataset (train_folders) that you are using. I’d advise you to create your custom dataset class. That way you can also return the path of the image and the problem is solved.
tutorial 54