nn.ModuleList
ModuleList:顾名思义,专门用于存储module的list。
参数
nn.ModuleList接受的必须是subModule类型,即不管ModuleList包裹了多少个列表,内嵌的所有列表的内部都要是可迭代的Module的子类 ,如:
nn.ModuleList([nn.ModuleList([Conv(inp_dim + j * increase, oup_dim, 1, relu=False, bn=False) for j in range(5)]) for i in range(
原创
2022-04-14 22:56:05 ·
1797 阅读 ·
1 评论