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

is there any way to convert other pytorch transformers to mlmodel? #4

Open
harold1505 opened this issue Aug 1, 2019 · 2 comments
Open

Comments

@harold1505
Copy link

Here the model generation shows how to convert gpt2 model specifically to mlmodel. How to apply this to other models like pretrained bert and xlnet? please help.

@julien-c
Copy link
Member

julien-c commented Aug 1, 2019

Hi, you would have to either

  • write a similar converter for another model (all those models are similar so the code wouldn't be so different),
  • or export to onnx then use onnx-coreml. (onnx-coreml does not fully support CoreML 3 right now but it's being worked on actively)

@harold1505
Copy link
Author

Thanks a lot for your reply. I have tried onnx-coreml but it doesn't provide support for all layers. I will try to write a converter like the one provided in the example.

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