基于pytorch实现手写数字识别(附python代码)

/1加载图片：加载数据集，没有的话会自动下载，数据分布在0附近，并打散。python 训练集：测试集=6k:1k。app utils.py文件：plot_image()绘制loss降低曲线； plot_curve()显示图片经过plot_image()可视化结果。minst_train.py文件：读取Minst数据集函数 /2 加载模型：三层线性模型，前两层用ReLU函数，batch_size=5

>>阅读原文<<