1、前言
本文基于Pytorch搭建CNN,训练MNIST手写数字数据集,代码完整、注释详细,新手直接运行。
2、数据集介绍
MNIST:7万张手写数字图片,28*28灰度图,深度学习入门通用数据集。
3、核心代码
python
import torch
import torchvision.models as models
4、训练结果
训练10轮准确率可达98%以上,拟合效果优秀。
5、总结
建议手动运行一遍,感受神经网络训练全过程。