使用PyTorch實現簡單的卷積神經網絡識別MNIST手寫數字

Pytorch 版本 0.4.1,系統:Ubuntu 16.04,Python版本:3.6 最近開始入門PyTorch,用了幾天的時間學習,寫了一個簡單的手寫數字識別練練手,參考了一個github的代碼 如今作一個小小的總結和記錄git 導入模塊及定義全局變量 程序使用到的模塊有:github import torch from torch import nn, optim # 網絡及優化器 im
相關文章
相關標籤/搜索