AttributeError: ‘list‘ object has no attribute ‘size‘解决办法

---

问题描述

在用torchsummary打印模型结构的时候报错，AttributeError: ‘list’ object has no attribute ‘size’。

解决办法

找到E:\program\anaconda\envs\dai\Lib\site-packages\torchsummary\torchsummary.py文件。

在第19行找到summary[m_key]["input_shape"] = list(input[0].size())

将这一句改成：

if isinstance(input[0], torch.tensor):
                summary[m_key]["input_shape"] = list(input[0].size())
            elif isinstance(input[0], list):
                summary[m_key]["input_shape"] = list(np.array(input[0].size()))

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

#修改前
summary[m_key]["input_shape"] = list(input[0].size())
#修改后
if isinstance(input[0], torch.Tensor):
   summary[m_key]["input_shape"] = list(input[0].size())
elif isinstance(input[0], list):
    summary[m_key]["input_shape"] = list(np.array(input[0]).shape)

成功！