pytorch中model.modules()和model.children()的區別


model.modules()和model.children()均為迭代器,model.modules()會遍歷model中所有的子層,而model.children()僅會遍歷當前層。

# model.modules()類似於 [[1, 2], 3],其遍歷結果為:
[[1, 2], 3], [1, 2], 1, 2, 3
 
# model.children()類似於 [[1, 2], 3],其遍歷結果為:
[1, 2], 3

用model.children()進行初始化參數時,可能會漏掉部分,用model.modules()會遍歷所有層

參考鏈接:https://discuss.pytorch.org/t/module-children-vs-module-modules/4551


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM