PP-DocLayoutV2 模型转onnx失败

#70
by Fiona1019 - opened

image

PaddlePaddle org

PP-DocLayoutV2目前暂不支持转ONNX格式

看着是因为argsort算子onnx不支持,但是可以用topK平替,有两种方法可以尝试:

  1. 编辑paddle 模型,在paddlepaddle框架内修改图结构,做算子替换
  2. 使用paddle2onnx的custom_ops,先转成onnx,然后编辑onnx替换argsort算子,但是最新版paddle2onnx去掉了custom_ops参数,这一条路应该走不通

Sign up or log in to comment