基於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

>>阅读原文<<