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