dvsseed / PyTorch_CNN_MNIST

PyTorch_CNN_MNIST (use Google colab with GPU)

CNN--PyTorch Coding Step:

1. 安裝套件
2. 匯入套件
3. 建立日誌檔
4. 讀取參數序列檔
5. 使用GPU(加速)或CPU
6. 圖片轉換(預處理)
7. 準備資料集(MNIST)：Train(42,000)、Validation(18,000)、Test(10,000)
8. 建立CNN模型(3Conv+3FC)
9. 設定CNN參數(Epochs, Learning Rate, Loss function, Optimizer)
10. 訓練(checkpoint)CNN模型(計算Accuracy, Loss並顯示結果)
11. 預測CNN模型(計算Accuracy, Loss並顯示結果)
12. 評估CNN模型-驗證指標(Validation index): 混淆矩陣(Confusion Matrix)
13. 調參優化
14. 儲存(checkpoint)模型相關數據
15. 讀取(checkpoint)模型相關數據
16. 測試CNN模型(計算Accuracy, Loss並顯示各類別的結果)
17. 讀取日誌

---