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
相关文章
相关标签/搜索