The file “image_classification _mlp.py” trains the Multilayer Perceptron (MLP) model on MNIST. This repository is an MLP implementation of classifier on MNIST dataset with PyTorch. The processes of my code can be broken down into following steps:
- Define the Network Architecture
- Define the training & test function for the Network
- Load and Visualize the data
- End2end training and testing pipeline
The file “image_classification _cnn.py” trains the Convolution Neural Network (CNN). Similarly, this repository is an CNN implementation of classifier on MNIST dataset with PyTorch. The processes of my code (same as previous above) can be broken down into following steps:
- Define the Network Architecture
- Define the training function and the test function
- Load and Visualize the data
- End2end training and testing pipeline
The attached file is the image classification summary report