Custom data loader and label encoding with CIFAR-10

Thanks,
Had several other issues,
I have a basic working version here:
https://github.com/QuantScientist/Deep-Learning-Boot-Camp/blob/master/day%2002%20PyTORCH%20and%20PyCUDA/PyTorch/19%20PyTorch%20CIFAR-10.ipynb

Now if my batch size is larger than 1, the images wont display :slight_smile: