Solved by beneyal in post #2 
              
              
                CIFAR10 is just an implementation of a torch.util.data.Dataset, and such a dataset needs only two methods: __len__ and __getitem__, as can be seen here: https://github.com/pytorch/vision/blob/master/torchvision/datasets/cifar.py 
Copy what you need from the CIFAR10 implementation and change the sour…