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