Train and Test CNN Model - Deep Learning with PyTorch 17