Pytorch實現mnist手寫數字識別

2020/6/29 Hey,突然想起來之前做的一個入門實驗,用pytorch實現mnist手寫數字識別。可以在這個基礎上增加網絡層數,或是嘗試用不同的數據集,去實現不一樣的功能。 Mnist數據集如圖: 代碼如下: import torch   import torch.nn as nn   import torch.utils.data as Data   import torchvision 
相關文章
相關標籤/搜索