地平线模型转换报错 ERROR The opset version of th..., only model with opset_version 10/11 is supported.

郝春妮

关注

阅读 70

2023-01-13


加上 opset_version=11

import torch.onnx
import torchvision
dummy_input = torch.randn(1, 3, 224, 224)
model = torchvision.models.alexnet(pretrained=True)
torch.onnx.export(model, dummy_input, "alexnet.onnx",opset_version=11)


精彩评论(0)

0 0 举报