Pytorch——初探onnx(1)解决upsample_bilinear2d转换问题
问题
将pytorch的pth文件读取后转换为ONNX格式,尝试使用下述代码将model导出为onnx格式
onnx.export(model, input_image, 'model.onnx')
出现了如下错误
RuntimeError: ONNX export failed: Couldn‘t export operator aten::upsample_bilinear2d
环境
目前测试发现pytorch1.6及以下均存在此问题
复现
下述代码中使用的mode为bilinear,align_co
原创
2021-07-07 14:21:52 ·
4721 阅读 ·
0 评论