Solved by Oli in post #2 
              
              
                Hey, I wrote this dataset for you that gets a subset of CIFAR10 . Just set the n_images_per_class to 500 and it should be ready to use! 
from torchvision import datasets
from collections import defaultdict, deque
import itertools


class Cifar5000(datasets.CIFAR10):
    def __init__(self, path, tran…