- PyTorch教程:21个项目玩转PyTorch实战
- 王飞 何健伟 林宏彬 史周安编著
- 103字
- 2025-03-28 14:38:53
1.5 总结
本章首先介绍了MNIST数据集,以及利用PyTorch导入该数据集,接着通过搭建神经网络的步骤(定义神经网络、前向传播、计算损失、反向传播、更新参数)构建出一个简单的图像卷积网络,最后对数据集进行训练并观察模型预测效果。
本章首先介绍了MNIST数据集,以及利用PyTorch导入该数据集,接着通过搭建神经网络的步骤(定义神经网络、前向传播、计算损失、反向传播、更新参数)构建出一个简单的图像卷积网络,最后对数据集进行训练并观察模型预测效果。