0
点赞
收藏
分享

微信扫一扫

Ubuntu 安装 Watt-Toolkit

本文重点

本文个人认为是本专栏最重要的章节内容之一,前面我们学习了pytorch中的基本数据tensor,后面我们就将学学习深度学习模型的内容了,在学习之前,我们先来看一下我们使用pytorch框架训练神经网络模型的基本步骤,然后我们下面就将这些步骤分解开来,详细学习。

代码

这个代码做了什么?

这个代码虽然仅仅有29行,但是确是深度学习模型训练的全部精髓,我们只要把这些掌握,那么毫无疑问pytorch就很简单了,我们先来简单的解释一下这几行代码做了什么?

1.首先引入了程序所需要的库

2.建立网络模型

建立网络模型需要注意几点:首先必须继承nn.Module,然后再__init__方法内构造我们的模型,最后forward方法中写上该网络模型的前向传播过程。

3.数据准备

本程序中构建了一个样本,数据为x,标签为y

4.创建网络模型实例

这个就是初始化网络模型对象就ok

5.定义损失函数

损失函数有很多根据自己的需求来定义,这里使用的是平方和损失函数

6.定义优化器

优化器有很多,我们常用的是随机梯度下降优化器SGD

7.前向传播

举报

相关推荐

0 条评论