tract icon indicating copy to clipboard operation
tract copied to clipboard

Accessing ONNX model metadata

Open jsgonsette opened this issue 3 years ago • 2 comments

Hello,

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

jsgonsette avatar Mar 20 '23 15:03 jsgonsette

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...

kali avatar Mar 20 '23 15:03 kali

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.

jsgonsette avatar Mar 23 '23 16:03 jsgonsette