在将PyTorch模型转换为ONNX模型时遇到警告，涉及shape推断问题。使用`F.interpolate`函数时出现问题，提供了两种解决方案：一是考虑用反卷积替代上采样；二是调整ONNX导出函数的opset_version参数为11。
在将pytorch模型转换为onnx模型时报警：
```
WARNING: The shape inference of prim::Constant type is missing, so it may result in wrong shape inference for the exported graph. Please consider adding it in symbolic function.
```
经过模块尝试发现是`F.interpolate`函数（换成F.upsmple也报）导致的，下面给出两种解决方法：
方法1. 用反卷积代替上采样（只是猜测，未尝试）
方法2. 将onnx.export函数的属性加上opset_version=11。（我自己用的这个方法）
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
