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

Export model to ONNX #4

Open
marlon-br opened this issue Aug 24, 2021 · 2 comments
Open

Export model to ONNX #4

marlon-br opened this issue Aug 24, 2021 · 2 comments

Comments

@marlon-br
Copy link

Hello, do you have any idea how to export the models to the onnx format?

I tried a script from huggingface: https://github.com/huggingface/transformers/blob/master/src/transformers/convert_graph_to_onnx.py

but it gives an error:
Error while converting the model: __init__() got an unexpected keyword argument 'feature_extractor'

@jonatasgrosman
Copy link
Owner

Hi @marlon-br ! Sorry for the very late reply. I had some health issues that forced me to stay offline for a while. My life is going back to normal slowly now...

I really don't know why it was happening to you. I hope you've already fixed that by yourself.
This repository is now deprecated in favor of https://github.com/jonatasgrosman/huggingsound. So you can try to reproduce this error on that new tool, and if it's still happening, please open an issue there, and I'll try to help you to fix it.

@chiaki-luo
Copy link

Hi @marlon-br ! Sorry for the very late reply. I had some health issues that forced me to stay offline for a while. My life is going back to normal slowly now...

I really don't know why it was happening to you. I hope you've already fixed that by yourself. This repository is now deprecated in favor of https://github.com/jonatasgrosman/huggingsound. So you can try to reproduce this error on that new tool, and if it's still happening, please open an issue there, and I'll try to help you to fix it.

Dear author, I want to make the onnx file with some hidden states outputs, but I don't have the architecture of module.nn, what should I do. I will appreciate it if you could give some related file.

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

3 participants