Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

paddle2onnx转基于rocketqa-zh-base-query-encoder训练后的模型报错:operators not supported diag_v2,kldiv_loss, #1194

Open
471417367 opened this issue Feb 21, 2024 · 0 comments
Labels
Bug Something isn't working Operator(New)

Comments

@471417367
Copy link

paddle2onnx-1.1.0

基于rocketqa-zh-base-query-encoder在自己的句子对上训练后动转静没有问题,但是在转onnx时报错。
paddle2onnx --model_dir ./checkpoint/model_32000 --model_filename inference.pdmodel --params_filename inference.pdiparams --save_file rocketqa32.onnx --enable_onnx_checker True --opset_version 16

[Paddle2ONNX] Start to parse PaddlePaddle model...
[Paddle2ONNX] Model file path: ./checkpoint/model_32000/inference.pdmodel
[Paddle2ONNX] Paramters file path: ./checkpoint/model_32000/inference.pdiparams
[Paddle2ONNX] Start to parsing Paddle model...
[Paddle2ONNX] Oops, there are some operators not supported yet, including diag_v2,kldiv_loss,
[ERROR] Due to the unsupported operators, the conversion is aborted.
Aborted (core dumped)

请问diag_v2,kldiv_loss 这两个算子是还没有实现了,有什么解决办法吗?

@Zheng-Bicheng Zheng-Bicheng added Bug Something isn't working Operator(New) labels Feb 27, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Bug Something isn't working Operator(New)
Projects
None yet
Development

No branches or pull requests

2 participants