CNNDemo testing pytorch with cnn, inspired by https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html