8000 onnx · Issue #5 · djukicn/loca · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

onnx #5

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
Open
lcx1874000 opened this issue Jan 17, 2024 · 1 comment
Open

onnx #5

lcx1874000 opened this issue Jan 17, 2024 · 1 comment

Comments

@lcx1874000
Copy link

Hello author, there was a problem converting the onnx modelraise RuntimeError("Unsupported: ONNX export of convolution for kernel "
RuntimeError: Unsupported: ONNX export of convolution for kernel of unknown shape.
how should this situation be resolved,Can l provide conversion onnx scripts?
@djukicn
Copy link
Owner
djukicn commented Jan 17, 2024

Hi @lcx1874000, I'm not an expert on ONNX, but it seems that it doesn't like lines 97 -- 99 in loca.py because of the dynamic size. Explicitly providing the size of prototypes seems to solve the problem. Of course, you will have to do this manually for every model you want to export, but if you just want to export the provided pretrained models, adding prototypes = prototypes.view(768, 1, 3, 3) to line 100 should help.

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
@djukicn @lcx1874000
0