Chinese-CLIP 在将rn50模型架构转换为onnx时,会报以下错误,

s6fujrry  于 2个月前  发布在  其他
关注(0)|答案(1)|浏览(34)

这个错误是因为在将rn50模型转换为ONNX格式时,遇到了不支持的属性"axes"。要解决这个问题,你可以尝试以下方法:

  1. 检查你的模型是否使用了不支持的属性"axes"。如果是这样,你需要修改模型以移除或替换该属性。

  2. 如果你确定需要使用"axes"属性,可以尝试更新PyTorch和torchvision库到最新版本,因为新版本可能已经支持该属性。你可以使用以下命令进行更新:

pip install --upgrade torch torchvision
  1. 如果问题仍然存在,你可以尝试在转换模型时禁用某些特性。例如,如果你使用的是torch.onnx.export()函数,可以尝试添加opset_version=11参数来指定ONNX的版本。示例代码如下:
import torch
import torchvision

# 加载模型

model = torchvision.models.resnet50(pretrained=True)
model.eval()

# 创建一个虚拟输入,用于模型转换

dummy_input = torch.randn(1, 3, 224, 224)

# 导出模型到ONNX格式,禁用某些特性

torch.onnx.export(model, dummy_input, "model.onnx", opset_version=11)

希望这些建议能帮助你解决问题。

p3rjfoxz

p3rjfoxz1#

您可以在ONNX官方网站上下载ONNX模型。

相关问题