pytorch转onnx遇到的问题汇总

  • Post author:
  • Post category:其他




1、关于torch导出onnx时候无法导出 upsample_bilinear2d的问题:

有人说直接将bilinear换成nearest,但是模型效果自然打折扣

完美的解决方案如下

torch.onnx.export(model, dummy_tensor, output, export_params=True,opset_version=11)

这里只需要在后面多加一个 opset_version = 11 就可以解决

但是前提是你的pytorch版本需要大于1.3.0



2、ONNX export failure: ONNX export fail



版权声明:本文为juluwangriyue原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。