About Data Loading

[V1oletM]

Hi, Thanks a lot for your code. I'm still confused about loading other datasets. For example, if I want to load cifar-10, would I have to rewrite the class CIFAR10 like base folder, url, init and so on in CIFAR100? If I want to load imagenet10 and imagenetdogs, would I have to add class ImageNet10 and ImageNetDogs which uses torchvision.datasets.ImageFolder to load file in pica/datasets.py and add datasets.regiser('imagenet10', ImageNet10)?

Thank you!