Pytorch Sequential 和 ModuleList 的区别

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

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

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

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

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

发布了218 篇原创文章 · 获赞 191 · 访问量 8万+

猜你喜欢

转载自blog.csdn.net/CSerwangjun/article/details/103286580
今日推荐