任务:用PyTorch实现多层网络(给代码截图参考)
引入模块,读取数据
构建计算图(构建网络模型)
损失函数与优化器
开始训练模型
对训练的模型预测结果进行评估
实现:PyTorch搭建多层全连接神经网络实现MNIST手写数字识别分类(参考:https://blog.csdn.net/out_of_memory_error/article/details/81414986)
1.全连接神经网络(FC)。输入层三个神经元,隐藏层四个神经元,输出层两个神经元。4
2.定义一个简单的三层全连接神经网络,每一层都是线性的
在每层输出部分添加激活函数
添加一个加快收敛速度的方法
测试模型