MNIST Training and Visualization

This is a toy sample for mnist classification using CNN and FCN.

Dependencies:

1. Python 3.6
2. Pytorch 0.4.0
3. matplotlib 2.2.2
4. numpy 1.14.3
5. Jupyter Notebook

Files:

1. MNIST.ipynb is a simple CNN network for classification in mnist dataset.
2. FCN.ipynb is mnist number classification using FCN network.
3. mnist_module.mat is .mat file to save the training CNN model.