解决用summary.py打印模型结构时报错AttributeError: ‘NoneType‘ object has no attribute ‘size‘

问题

在使用pytorch想打印自己的网络模型结构，结果报错AttributeError: ‘NoneType’ object has no attribute ‘size’
Ctrl+鼠标左键进入torchsummary函数里面

解决方案

经过网上调查资料，发现是因为torchsuammry版本太高了，我用到的版本是1.5左右，因此尝试换一个低版本的，比如1.4。
先卸载原来的：

pip uninstall torchsummary

再安装一个低版本的，1.4左右的都可以用，这个安装速度很快，也不用镜像源，如果需要的话可以在后面加个-i https://pypi.tuna.tsinghua.edu.cn/simple some-package。

pip install torchsummary == 1.4.4

然后就可以成功打印啦！