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.

Sign up for GitHub

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

Jump to bottom

转onnx问题 #122

Open
sunzx8 opened this issue Mar 5, 2024 · 1 comment
Open

转onnx问题 #122

sunzx8 opened this issue Mar 5, 2024 · 1 comment
Assignees
@wangyuxinwhy
Labels
enhancement New feature or request

Comments

@sunzx8
Copy link

sunzx8 commented Mar 5, 2024

🚀 The feature

在用optimun转换为onnx之后，我发现输出多了一个：
image
请问为什么会这样？您给出的推理示例直接输入了整个句子，我在构建输入的时候是只用给input_id 和attention_mask吗？token_type_ids需不需要输入？
@sunzx8 sunzx8 added the enhancement New feature or request label Mar 5, 2024
@wangyuxinwhy
Copy link
Owner

token_type_ids 是不需要输入的，输出多个也是正常的，token_embeddings 可以忽略掉（是每个 token 的编码），用 sentence_embedding 就可以了(是整句话的编码)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@wangyuxinwhy wangyuxinwhy

Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@wangyuxinwhy @sunzx8