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

说明

因为最近刚开始学pytorch，想输出模型结果来看看，但是他并没有像keras那么简单，就挺苦恼的。但学习的过程从来都不会一帆风顺的，加油吧。

使用代码

报错截图

查明原因

首先找到上面截图的红框框，找到torchsummary.py文件中的那一行代码
我们输出一下input[0]的类型


发现前面传进去的都是torch.Tensor类型，最后一个是list，学python的都知道list是没有size属性的，这确实。（这里如果有大佬能解决：为什么最后一个传进去的类型是list的话，应该就可以不用修改以下代码了吧估计）

解决方法

反正最终要赋给summary的值是list的形式，即[batch_size,channel,height,width]
那就来个判断，如果是torch.Tensor的话，那就还是按照源码这样写；如果是list类型的话，那就先将他转成numpy.array形式，因为数组是有个shape属性的。

成功输出模型

最后

初学pytorch的小白，如果有什么错误的话请谅解并提出来。
还有就是如果有大佬能解决上面所说的，为什么最后一个传进去的类型是list而不是torch.Tensor的话，请告诉我，谢谢啦。