我是Pytorch的新手,我不太明白的一件事就是使用nn.ModuleList和nn.Sequential.我可以知道何时应该使用一个而不是另一个？谢谢.
Ego*_*kin 12
nn.ModuleList没有forward方法,但是nn.Sequential确实有它.因此,您可以在nn.Sequential中包装几个模块并在输入上运行它.
nn.ModuleList只是一个Python列表(尽管它很有用,因为参数可以通过优化器发现和训练),nn.Sequential是一个顺序运行输入组件的模块.
希望有所帮助.
| 归档时间： |
|
| 查看次数： |
2182 次 |
| 最近记录： |