pytorch學習3-使用全鏈接神經網絡進行MNIST數據集分類

1.模型使用python 使用一層輸入層,一層隱藏層,一層輸出層的簡單全鏈接神經網絡,激活函數採用relu函數算法 2.代碼以下網絡 # -*- coding: utf-8 -*- """ Created on Mon Apr 29 10:12:03 2019 @author: hp """ import torch import torch.nn as nn import torchvisi
相關文章
相關標籤/搜索