pytorch 转 Onnx（OCR 模型）
pytorch to Onnx(OCR model)
我正在尝试在给定的 repo 中转换 pytorch 模型 https://github.com/clovaai/deep-text-recognition-benchmark
到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
Link 到 git 问题 https://github.com/clovaai/deep-text-recognition-benchmark/issues/76
有什么建议吗？
谢谢。
adaptive_avg_pool2d 在我的案例中不受支持，这个 nn.AdaptiveAvgPool2d((None,1)) 也有问题。
图层
nn.AdaptiveAvgPool2d((None,1)) .
None 实际上是导致错误的原因；我们需要让它静态化来解决错误。您可以将 'None' 更改为静态值。例如：
nn.AdaptiveAvgPool2d((512,1)).
我正在尝试在给定的 repo 中转换 pytorch 模型 https://github.com/clovaai/deep-text-recognition-benchmark 到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
Link 到 git 问题 https://github.com/clovaai/deep-text-recognition-benchmark/issues/76
有什么建议吗？
谢谢。
adaptive_avg_pool2d 在我的案例中不受支持，这个 nn.AdaptiveAvgPool2d((None,1)) 也有问题。
图层
nn.AdaptiveAvgPool2d((None,1)) .
None 实际上是导致错误的原因；我们需要让它静态化来解决错误。您可以将 'None' 更改为静态值。例如：
nn.AdaptiveAvgPool2d((512,1)).