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

UIE多输入转onnx后输入结构与原模型不符 #1282

Open
XiHenSuper opened this issue Jun 14, 2024 · 1 comment
Open

UIE多输入转onnx后输入结构与原模型不符 #1282

XiHenSuper opened this issue Jun 14, 2024 · 1 comment

Comments

@XiHenSuper
Copy link

XiHenSuper commented Jun 14, 2024

问题描述
UIE原始模型的输入有三个: input_ids、position_ids以及常量fill constant
通过paddle2onnx.command.c_paddle_to_onnx函数转onnx后输入结构发生大的变动。

更多信息 :

  • 用于部署的推理引擎: onnx
  • 为什么需要转换为ONNX格式:需要适配国产化系统
  • Paddle2ONNX版本: 1.1.0
  • 你的联系方式(Email/Wechat/Phone):

报错截图
UIE原始输入模型结构:
image

转onnx后的输入模型结构:
image

其他信息

@Jiang-Jia-Jun
Copy link
Collaborator

与原模型哪里不符?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants