search

elixir-nx / ortex

ONNX Runtime bindings for Elixir
MIT License
122 stars 15 forks source link

Add example for BERT sequence classification #30

Closed victor23k closed 3 months ago

victor23k commented 3 months ago

Hello! :heart: the library, I'm using it right now to run inference on a BERT fine-tuned for sequence classification.

@mortont do you think it would be helpful to add an example on how to set up the pipeline for inference similar to the StableLM example? The main thing that I did not find documented anywhere (maybe I didn't looked hard enough) was getting the id2label mappings from the generated config.json. If you think it will add value, I'll prepare the PR.

mortont commented 3 months ago

Thanks for the feedback! Yes a PR would definitely be welcome, anything you found undocumented along the way could help others get started.