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 trained model for Tensorflow/PyTorch/C++? #453

Open
liuyxpp opened this issue Nov 1, 2023 · 2 comments
Open

Export trained model for Tensorflow/PyTorch/C++? #453

liuyxpp opened this issue Nov 1, 2023 · 2 comments
Labels
enhancement New feature or request

Comments

@liuyxpp
Copy link

liuyxpp commented Nov 1, 2023

Is there any documented way to export the trained Lux model to be loaded by Tensorflow/PyTorch or even C++? Is ONNX the correct way to go? If so, how can I export Lux model as ONNX?
@avik-pal
Copy link
Member

avik-pal commented Nov 1, 2023

I don't think there is currently any way to export models from Lux (or more broadly julia) to ONNX. the other way is possible via https://github.com/DrChainsaw/ONNXNaiveNASflux.jl -> Flux Model -> Lux.transform

@liuyxpp
Copy link
Author

liuyxpp commented Nov 3, 2023 •
edited
Loading

Just scanned the source code of ONNXNaiveNASflux.jl which seems it supports exporting flux model to ONNX in src/serialize/serialize.jl? Or do I miss anything?

@avik-pal avik-pal added the enhancement New feature or request label May 1, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@liuyxpp @avik-pal