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

export onnx #101

Open
Awesome0324 opened this issue Mar 19, 2024 · 1 comment
Open

export onnx #101

Awesome0324 opened this issue Mar 19, 2024 · 1 comment

Comments

@Awesome0324
Copy link

Hello, I encountered some issues while exporting MSA to onnx.

I used MSA to train the REFUGE dataset and set multiask_output to 2 as required, successfully saving checkpoint best. pth.

But when I wanted to export checkpoint_best.pth as onnx, it prompted me that the forward parameter in torch's onnx was missing the multiask_output parameter. I would like to inquire about how to solve this problem.
@puallee
Copy link

puallee commented Apr 19, 2024

can you give me more detail？

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@puallee @Awesome0324