什么时候应该使用nn.ModuleList,什么时候应该使用nn.Sequential?
When should I use nn.ModuleList and when should I use nn.Sequential?
我是 Pytorch 的新手,我不太明白的一件事是 nn.ModuleList
和 nn.Sequential
的用法。我能知道什么时候应该使用一个吗?谢谢
nn.ModuleList
does not have a forward
method, but nn.Sequential
确实有一个。因此,您可以将多个模块包装在 nn.Sequential
中,并在输入中 运行 包装。
nn.ModuleList
只是一个 Python list(尽管它很有用,因为可以通过优化器发现和训练参数)。 nn.Sequential
是一个按顺序 运行 输入组件的模块。
我是 Pytorch 的新手,我不太明白的一件事是 nn.ModuleList
和 nn.Sequential
的用法。我能知道什么时候应该使用一个吗?谢谢
nn.ModuleList
does not have a forward
method, but nn.Sequential
确实有一个。因此,您可以将多个模块包装在 nn.Sequential
中,并在输入中 运行 包装。
nn.ModuleList
只是一个 Python list(尽管它很有用,因为可以通过优化器发现和训练参数)。 nn.Sequential
是一个按顺序 运行 输入组件的模块。