Solved by Kushaj in post #12 
              
              
                Here is an alternative. Use fastai_datasets to get your CIFAR100 dataset. The reason being it provides data in Imagenet form i.e. every class/label image are in their separate folders and you can just delete the folders/classes that you do not want. 
To create your dataset you would use ImageFolder,…