pytorch-summary 針對DenseNet生成摘要報錯: AttributeError: 'list' object has no attribute 'size'


pytorch-summary 針對DenseNet生成摘要報錯:  AttributeError: 'list' object has no attribute 'size' 

google搜索這個問題,發現也有人遇到過:https://github.com/sksq96/pytorch-summary/issues/2

作者說這個問題已經被解決了,但是我還遇到了,那多半就是版本的問題。

我裝的版本是1.4.5,切換成1.4.4,成功解決!

 pip install torch-summary==1.4.4

 


免責聲明!

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



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