python 属性错误:"collections. OrderedDict"对象没有"modules"属性

p4tfgftt  于 2023-01-19  发布在  Python
关注(0)|答案(1)|浏览(1021)

我尝试使用以下代码将pt文件导出到onnx:

import torch import torchvision

dummy_input = torch.randn(10, 3, 224, 224, device="cuda") model = torch.load('base_40m_textvec.pt')

input_names = [ "actual_input_1" ] + [ "learned_%d" % i for i in range(16) ] output_names = [ "output1" ]

torch.onnx.export(model, dummy_input, "base_40m_textvec.onnx", verbose=True, input_names=input_names, output_names=output_names)

但是,我不断得到这个错误:

AttributeError: 'collections.OrderedDict' object has no attribute 'modules'

我试着在互联网上搜索错误,我得到了以下结果

object has no attribute 'eval'
object has no attribute 'parameters'
object has no attribute 'value_counts'

object has no attribute 'modules'没有
这是我第一次尝试转换到onnx,所以我不知道为什么它不工作。任何帮助?

yhxst69z

yhxst69z1#

您加载的检查点“base_40m_textvec.pt”可能是OrderedDict,而不是torch.nn.Module

相关问题