pytorch 使用summary()报错，解决torchsummary进行模型评估测试报错‘dict‘ object has no attribute ‘size‘

pytorch 使用summary(model, (2,32,32))打印模型报错。

 解决方法：

根据报错路径找到torchsummary.py文件，打开torchsummary.py，

找到21行if isinstance(output, (list, tuple)):

在第21行if isinstance(output, (list, tuple)):之前加入以下代码将dict中value转化为list类型

if isinstance(output, dict):
    output = list(output.values()

再次运行程序就可以打印出模型参数了

亲测有效。 

参考解决torchsummary进行模型评估测试报错‘dict‘ object has no attribute ‘size‘_specializeddemo8的博客-CSDN博客