pytorch导出onnx的时候报以下错误:
原因是代码中使用interpolate进行了缩放
F.interpolate(x, size=(512, 512), mode='bilinear', align_corners=False)
解决办法:
opset_version=11
成功导出成onnx
pytorch导出onnx的时候报以下错误:
原因是代码中使用interpolate进行了缩放
F.interpolate(x, size=(512, 512), mode='bilinear', align_corners=False)
解决办法:
opset_version=11
成功导出成onnx