0
点赞
收藏
分享

微信扫一扫

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


加上 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 条评论