Pytorch Sequential 和 ModuleList 的区别

Sequential 和 ModuleList都是一种特殊的Module.

可以理解成Module的列表或者序列。

相同:都是存储了多个module。

区别:Sequential里的module会将输入一层一层的传下去,在深度网络的实现中,一层网络的输出是下一层网络的输入是很常见的,这时候用Sequential比较好,它可以将上层的输出自动传入下一层的输入,不用我们自己去实现,命名变量啥的。

问题:为什么不用Python的List而用ModuleList, 因为ModuleList里的module会被自动上层module自动识别为子module,而Python自带的List不行。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章