在代码中我们经常看到先将网络结构添加到列表中,再转化为ModuleList类型
代码示例:
import torch
import torch.nn as nn
#建立列表
convs=[]
l=nn.Conv2d(3,3,kernel_size=1)
convs.append(l)
#转化为ModuleList类型
module=nn.ModuleList(convs)
print(list(module.modules()))
输出:
[ModuleList(
(0): Conv2d(3, 3, kernel_size=(1, 1), stride=(1, 1))
), Conv2d(3, 3, kernel_size=(1, 1), stride=(1, 1))]