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

原因：网络为多头输出，input[0]的内容为list(第一个输出、第二个输出、第三个输出...)。i

修改./lib/python3.7/site-packages/torchsummary/torchsummary.py中的代码

summary[m_key]["input_shape"].append(x.shape)

修改为如下所示：

if isinstance(input[0],list):
    summary[m_key]["input_shape"] = []
    for x in input[0]:
        summary[m_key]["input_shape"].append(x.shape)
    else:
        summary[m_key]["input_shape"] = list(input[0].size())