问题
在使用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
然后就可以成功打印啦!