PyTorch-CNN-MNIST

This repo contains a sample code to show how to create a cnn model using pytorch and trained on the mnist.

Demo

Install

Install pytorch and other necessary module (tensorboard and matplotlib).

pip install -r requirements.txt

Train

Run train.py to train your cnn model on mnist dataset.

python train.py

Test

Run test.py to test the trained model on your own handwritten digit.

python test.py

Deploy

1. register an account of heruku
2. create an app on heruku
3. push the deploy folder to heroku master

Cite

https://github.com/avinassh/pytorch-flask-api-heroku
https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html