pytorch 使用AlexNet实现Mnist手写数字识别

AlexNet网络框架以下：AlexNet的原始输入图片大小为224*224，Mnist数据集中图片大小为28*28，因此须要对网络参数进行修改。python 先掉用train函数进行训练，训练好的参数会保存在params.pth文件中，训练好使用本地图片（画图软件生成）进行测试。网络 完整程序以下：框架 import torch import torchvision as tv import t

>>阅读原文<<