pytorch入門_一個簡單實例_CNN:MNIST數據集手寫數字識別

將MNIST數據集手寫數字識別樣例做爲模板: MNIST 包括6萬張28x28的訓練樣本,1萬張測試樣本。Pytorch裏面包含了MNIST的數據集,直接使用便可。python import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torchvision
相關文章
相關標籤/搜索