使用torchsummary，报错AttributeError: ‘list‘ object has no attribute ‘size‘

1. 出现问题

在使用torchsummary对pytorch网络结构可视化时，出现如下错误：

最秀的是，运行完summary后，让模型对输入进行处理也会出现该错误：

 找了半天bug，发现原来是包安装错了

2. 解决问题

将torchsummary卸载后，安装torch-summary

pip uninstall torchsummary
pip install torch-summary

输出网络结构：

 这个bug真是有点迷