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