Pytorch学习 model类

model是如何作用的

model的两个要素

1.构建子模块: model的初始化函数 init(),要使用supper调用nn.model类,在初始化函数中我们构建子模块。
2. 拼接子模块: model 的 forward函数中我们用子模块构建计算图,返回计算结果

在model类的引用处设置断点

Pytorch学习 model类

步入leNet中
进入定义leNet的model文件中:
Pytorch学习 model类
模型构建第一个要素,构建子模块,就是在model文件,继承nn.model()类的自定义类中的__init__()中构建,在实例化阶段只会调用模型的__init__()模块。遍历每一个模块后,就跳出model文返回训练文件

接下里一直到训练阶段才会再次调用模型
Pytorch学习 model类
进入前向传播对应代码里
Pytorch学习 model类
可以看到进入model的call函数,我们手写的model是Pytorch自带的nn.model
的一个实例,如果这一步可以进入call函数,说明我们的model可以被调用。我们需要关注的是上述代码中的forward。
进入forward:
Pytorch学习 model类
就会进入一个到我们自己在model文件中编写的forward函数,在forward函数里进行前向传播运算。返回值赋给了outputs

可以看到Pytorch模型搭建离不开nn.model类,我们有必要了解一下nn.model类