解决torchsummary进行模型评估测试报错‘dict‘ object has no attribute ‘size‘

最新推荐文章于 2024-05-31 17:12:32 发布

原创最新推荐文章于 2024-05-31 17:12:32 发布 · 4.6k 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #pytorch #python

在使用torchsummary库进行VGG-16模型的评估时遇到报错，原因是尝试获取字典对象的size属性。解决方法是在torchsummary源码中添加代码，将字典值转换为list类型，然后重新运行模型评估，成功输出了模型摘要。

部署运行你感兴趣的模型镜像

使用torchsummary进行模型评估测试报错

- 问题：使用from torchvision.models.vgg import VGG引用VGG-16。作为FCN前置网络，在使用torchsummary进行模型评估测试报错
- 解决方案：

问题：使用from torchvision.models.vgg import VGG引用VGG-16。作为FCN前置网络，在使用torchsummary进行模型评估测试报错

//作为FCN前置网络，在使用torchsummary进行模型评估测试报错

Traceback (most recent call last):
  File "F:/ggsddu/graduationProject/tongue_segmentation/FCN.py", line 163, in <module>
    summary(model=fcn_model, input_size=(3, 224, 224), batch_size=-1, device="cpu")
  File "D:\anaconda\envs\pytorch1.2\lib\site-packages\torchsummary\torchsummary.py", line 72, in summary
    model(*x)
  File "D:\anaconda\envs\pytorch1.2\lib\site-packages\torch\nn\modules\module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "F:/ggsddu/graduationProject/tongue_segmentation/FCN.py", line 28, in forward
    output = self.pretrained_net(x)
  File "D:\anaconda\envs\pytorch1.2\lib\site-packages\torch\nn\modules\module.py", line 495, in __call__
    hook_result = hook(self, input, result)
  File "D:\anaconda\envs\pytorch1.2\lib\site-packages\torchsummary\torchsummary.py", line 26, in hook
    summary[m_key]["output_shape"] = list(output.size())
AttributeError: 'dict' object has no attribute 'size'

进程已结束，退出代码为 1

解决方案：

打开torchsummary.py，找到23行if isinstance(output, (list, tuple)):
在之前加入以下代码将dict中value转化为list类型

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

再运行成功输出模型
在这里插入图片描述

您可能感兴趣的与本文相关的镜像

PyTorch 2.8

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理