很经典的一个问题:使用pytorch实现mnist手写数字识别。 模型在测试集上的准确率是评判标准。 因此可以从以下角度来获取高准确率:
- 调整结构(包括神经网络层、学习率等)
- 在防止过拟合的前提下增加训练轮数
____.ipynb
是作业系统提供的,题目要求等信息在其中。
Untitled.ipynb
是新建的,来源于知乎文章[Pytorch]CNN识别手写数字(Mnist数据集),最高准确率99.71%,使用了残差神经网络,效果良好。最终实现了在模型上99.54的准确率。
main.py
用于上传作业系统,实现题目要求。
model_train.py
也是训练模型使用的。
2023.5.28