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