Image Classification

Basic neural network for image classification in PyTorch. The model takes about 2 hours to train with 10 epochs of 50000 images per epoch. I use training batches of 64. The model uses basic layers including CNN, Relu, max pooling, FNN, and a final Softmax layer. The performance of the model improves as it is trained, as expected.