文章目錄
- 1、簡介
- 2、三者之間的區別
- 3、如何讓nn.ModuleList 和nn.ModuleDict實現推理
- 3.1 方案1: 實現forward函數
- 3.2 方案2: 將nn.ModuleList 和nn.ModuleDict轉換為nn.Sequential
- 4、nn.ModuleDict、nn.ModuleList 的區別
- 5、nn.ModuleList 、 nn.ModuleDict 與 Python list、Dict 的區別
在使用pytorch搭建網絡的時候,經常會使用到
nn.Sequential
、
nn.ModuleDict
、
nn.ModuleList
這三個類, 這三個類都是繼承自
nn.Module
類,本文將詳細介紹這3個類的使用。