Digits-MNIST-Pytorch

Handwritten Digits classification based on MNIST dataset , implemented using Pytorch

About

The MNIST Handwritten digits dataset is one of the famous dataset used to image classification tasks. It consists of images belonging to each of the 10 digit classes.
Each image is of the size 28 x 28.

Our Model Architecture

CONVOLUTION LAYER --> MAX POOL --> CONVOLUTION LAYER --> MAXPOOL --> FC LAYER --> FC LAYER --> OUTPUT LAYER