Pytorch Sequential 和 ModuleList 的區別

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

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

相同:都是存儲了多個module。

區別:Sequential裏的module會將輸入一層一層的傳下去,在深度網絡的實現中,一層網絡的輸出是下一層網絡的輸入是很常見的,這時候用Sequential比較好,它可以將上層的輸出自動傳入下一層的輸入,不用我們自己去實現,命名變量啥的。

問題:爲什麼不用Python的List而用ModuleList, 因爲ModuleList裏的module會被自動上層module自動識別爲子module,而Python自帶的List不行。

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