pytorch - 什么时候应该使用 nn.ModuleList，什么时候应该使用 nn.Sequential？

Question

我是 Pytorch 的新手，我不太了解的一件事是 nn.ModuleList 的用法。和 nn.Sequential .我能知道什么时候应该使用一个而不是另一个吗？谢谢。

Answer 1

nn.ModuleList 没有 forward方法，但是 nn.Sequential 有一个。所以你可以在 nn.Sequential 中包裹几个模块并在输入上运行它。nn.ModuleList只是一个 Python 列表(尽管它很有用，因为可以通过优化器发现和训练参数)。虽然 nn.Sequential是一个在输入上顺序运行组件的模块。