nn.Sequential与nn.ModuleList的区别


简单的罗列一些重要的信息:

nn.ModuleList

  • 相当于Python中的list,当你想在PyTorch中使用list装载Module时,不能用list,而应该使用nn.ModuleList
  • 没有forward方法,只是一个容器

nn.Sequential

  • 按照顺序将Module们组织在一起,前一个Module的输出是后一个Module的输入;
  • 拥有forward方法,可以将其看做是一个Module直接使用

参考资料

  1. when-should-i-use-nn-modulelist-and-when-should-i-use-nn-sequential

文章作者: CarlYoung
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 CarlYoung !
  目录