Building a Convolutional Neural Network(CNN) Based on PyTorch to Train and Recognize the Handwritten Digit Dataset MNIST;
基于PyTorch搭建卷积神经网络(CNN),实现手写数字数据集Mnist的训练与识别
If you want to train, enter src
folder and run:
python train.py
The trained model will be saved in the model
folder.
If you want to test the performance of the trained model, you need to run it in the src
folder:
python test.py