Mix of images with own dataset

[eplatas]

Hi, thank you very much, your work is very interesting; I have a question regarding the datset, I'm working with the code that you uploaded in Colab, if I want to train with my own data, how do I modify the line of code?

I modify la line of torchvision, I chance CIFAR10 to ImageFolder and I structure the folders according to ImageNet but I'm still having problems.

testset = torchvision.datasets.ImageFolder(root="/root/test/all/", transform=transforms.Compose([transforms.ToTensor()]), target_transform=None).

I thank you very much for your time and attention

cheers