【发布时间】:2018-05-12 15:31:08
【问题描述】:
我是 Pytorch 的新手,我不太了解的一件事是 nn.ModuleList 和 nn.Sequential 的用法。我可以知道什么时候应该使用其中一种吗?谢谢。
【问题讨论】:
标签: pytorch
我是 Pytorch 的新手,我不太了解的一件事是 nn.ModuleList 和 nn.Sequential 的用法。我可以知道什么时候应该使用其中一种吗?谢谢。
【问题讨论】:
标签: pytorch
nn.ModuleList 没有forward 方法,但nn.Sequential 确实有一个。因此,您可以在 nn.Sequential 中包装多个模块并在输入上运行它。
nn.ModuleList 只是一个 Python list(尽管它很有用,因为可以通过优化器发现和训练参数)。而nn.Sequential 是一个在输入上顺序运行组件的模块。
【讨论】: