pytorch使用CNN實現MNIST手寫體的識別

**MNIST數據集介紹:**MNIST數據集在torvision.datasets裏面,能夠自行加載,其中訓練集有6W張,測試集有1W,都爲灰度圖,即channel爲1,圖片的大小都是28x28,下面圖片就是一部分數據。 web 導入工具包 import torch from torch.autograd import Variable import torch.nn as nn import
相關文章
相關標籤/搜索