#1.数据集说明 mnist：https://zhuanlan.zhihu.com/p/547147237?utm_id=0

#2.文件说明：

• checkpoints：模型文件存储位置
• data：数据集存储位置
• model：模型代码
• train.py: 训练代码
• predict.py：预测代码

#2.使用流程：

• 运行train.py进行训练

右键run，会使用默认值运行。

或者命令行传参运行：
python train.py --gpu 0 --epochs 50 --batch_size 64 --lr 0.1 --dataset_path ./data/

• 运行predict.py进行单个图片的预测

右键run，会使用默认值运行。

或者命令行传参运行：
python predict.py --gpu 0 --model_savepath ./checkpoints/mnist-lenet5.pth --dataset_path ./data/