pytorch 到 Onnx（OCR 模型）答案

【问题标题】：pytorch to Onnx(OCR model)pytorch 到 Onnx（OCR 模型）
【发布时间】：2020-06-02 12:37:49
【问题描述】：

我正在尝试在给定的 repo https://github.com/clovaai/deep-text-recognition-benchmark 中转换 pytorch 模型到onnx。

我在这样做时遇到了问题。

Failed to export an ONNX attribute 'onnx::Gather', since it's not constant, please try to make things (e.g., kernel size) static if possible

有什么建议吗？

谢谢。

【问题讨论】：

【解决方案1】：

图层

nn.AdaptiveAvgPool2d((None,1)).

实际上没有导致错误；我们需要将其设为静态以解决错误。您可以将“无”更改为静态值。例如：

nn.AdaptiveAvgPool2d((512,1)).

【讨论】：

【解决方案2】：

adaptive_avg_pool2d 在我的情况下不受支持，并且这个 nn.AdaptiveAvgPool2d((None,1)) 也有问题。

【讨论】：