首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

【Pytorch 】笔记四:Module 与 Containers 的源码解析

首先学习模型的创建步骤和 nn.Module 的相关细节, 然后学习搭建模型的容器Containers,这里面包括 nn.Sequential, nn.ModuleList, nn.ModuleDict...相当于一个运算, 必须实现 forward() 函数(从计算图的角度去理解) 每个 module 都有 8 个字典管理它的属性(最常用的就是_parameters,_modules ) 3.模型容器 Containers...上面我们学习的模型的搭建过程,包括两个要素:构建子模块和拼接子模块, 在搭建模型中,还有一个非常重要的概念,那就是模型容器 Containers。...Containers 这个容器里面包含 3 个子模块,分别是nn.Sequential, nn.ModuleList, nn.ModuleDict, 下面我们一一来看一看: 3.1 nn.Sequential...第二块是我们的模型容器 Containers 部分,这里面先学习了 nn.Sequential, 这个是顺序搭建每个子模块, 常用于 block 构建,依然是通过代码调试看了它的初始化和自动前向传播机制

98760
领券