pytorch 使用AlexNet實現Mnist手寫數字識別

AlexNet網絡框架以下：AlexNet的原始輸入圖片大小爲224*224，Mnist數據集中圖片大小爲28*28，因此須要對網絡參數進行修改。python 先掉用train函數進行訓練，訓練好的參數會保存在params.pth文件中，訓練好使用本地圖片（畫圖軟件生成）進行測試。網絡 完整程序以下：框架 import torch import torchvision as tv import t

>>阅读原文<<