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

Accessing ONNX model metadata #1006

Open
jsgonsette opened this issue Mar 20, 2023 · 2 comments
Open

Accessing ONNX model metadata #1006

jsgonsette opened this issue Mar 20, 2023 · 2 comments

Comments

@jsgonsette
Copy link

Hello,

I'm wondering if there is any mean to retrieve the metadata associated with a ONNX model from the Tract API ?

@kali
Copy link
Collaborator

kali commented Mar 20, 2023

Hello!

Thanks for your interest in tract.

Not at this stage. I guess it would not be super hard to implement though, as tract models have a property hash where pretty much anything could be put when parsing the onnx model...

@jsgonsette
Copy link
Author

Thanks for your interest in tract.

Thanks to you :-) I'm relatively new to Rust but I could easily open and run inferences on my Transformer model with Tract. Regarding how uneasy the deployment of NN on small applications or devices may be, it is very appreciable.

For my problem, the property field was indeed the place where I was expecting to find my metadata.
If this improvement is not too complex, it would be very nice to consider it in your backlog.
Using metadata is far more professional and secure than having to rely on convoluted file naming to remember where a model comes from.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants