Pytorch Cnn Model Example