当前位置: 代码迷 >> 综合 >> 关于 ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d 解决方案
  详细解决方案

关于 ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d 解决方案

热度:20   发布时间:2023-12-16 03:02:53.0

关于torch导出onnx时候无法导出 upsample_bilinear2d的问题:
有人说直接将bilinear换成nearest,但是模型效果自然打折扣
完美的解决方案如下

torch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11)
1
这里只需要在后面多加一个 opset_version = 11 就可以解决
但是前提是你的pytorch版本需要大于1.3.0

  相关解决方案