我应该何时使用nn.ModuleList,何时应该使用nn.Sequential?

时间:2017-11-29 02:24:37

标签: pytorch

我是Pytorch的新手,我不太了解的一件事是使用nn.ModuleListnn.Sequential。我可以知道何时应该使用一个而不是另一个?感谢。

1 个答案:

答案 0 :(得分:10)

nn.ModuleList没有正向方法,但nn.Sequential确实拥有它。因此,您可以在nn.Sequential中包装几个模块并在输入上运行它。

nn.ModuleList只是一个Python列表(虽然它很有用,因为参数可以通过优化器发现和训练),nn.Sequential是一个顺序运行输入组件的模块。

希望有所帮助。