PyTorch-Mnist

Building a Convolutional Neural Network(CNN) Based on PyTorch to Train and Recognize the Handwritten Digit Dataset MNIST;
基于PyTorch搭建卷积神经网络(CNN)，实现手写数字数据集Mnist的训练与识别

Train

If you want to train, enter src folder and run:

python train.py

The trained model will be saved in the model folder.

Test

If you want to test the performance of the trained model, you need to run it in the src folder:

python test.py