模型创建与nn.module
一、模型的创建
模型创建的步骤:
模型构建的两个要素
- 构建子模块
- 拼接子模块
1.1 构建子模块
以之前上传的纸币二分类代码为例,对模型的构建进行讲解。
在70行设置断点,查看如何构建LeNet网络。
step into 进入
进入到LeNet的__init__函数构建子模块,创建了两个卷积层与三个全连接层。运行完最后一个子模块self.fc3就会跳出返回,模型的初始化便完成了。
1.2 拼接子模块
模块在__init__构建好之后,神魔时候拼接实现前向传播呢。
下次