我是Pytorch的新手,我不太了解的一件事是使用nn.ModuleList
和nn.Sequential
。我可以知道何时应该使用一个而不是另一个?感谢。
答案 0 :(得分:10)
nn.ModuleList没有正向方法,但nn.Sequential确实拥有它。因此,您可以在nn.Sequential中包装几个模块并在输入上运行它。
nn.ModuleList只是一个Python列表(虽然它很有用,因为参数可以通过优化器发现和训练),nn.Sequential是一个顺序运行输入组件的模块。
希望有所帮助。