pth –> onnx报错
RuntimeError: ONNX export failed: Couldn't export operator aten::upsample_bilinear2d
解决:
torch.onnx.export(model, input, onnx_path, verbose=True, input_names=input_names, output_names=output_names)
#增加:opset_version=11
torch.onnx.export(model, input, onnx_path, verbose=True, input_names=input_names, output_names=output_names, opset_version=11)
版权声明:本文为weixin_40437821原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。