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 ·
4132 阅读 ·
0 评论