search

microsoft / onnxruntime-training-examples

Examples for using ONNX Runtime for model training.
MIT License
311 stars 62 forks source link

Train ONNX models with Rust. #136

Closed 9876691 closed 1 year ago

9876691 commented 1 year ago

How would I go about training an onnx model using the onnx runtime and Rust?

There is a Rust wrapper for onnx runtime https://github.com/pykeio/ort

Would this need to be extended?

baijumeswani commented 1 year ago

ONNX Runtime Training does not currently have any rust bindings yet.

C training api is available for consumption here.

ONNX Runtime now has rust bindings available at https://github.com/microsoft/onnxruntime/tree/main/rust. External contributions for rust bindings for training are welcome.

vtantia commented 1 year ago

Thanks @baijumeswani for your previous answer! Is it in the roadmap to have rust bindings for ONNX Runtime Training by any chance?

baijumeswani commented 1 year ago

@vtantia, the work for adding the Rust bindings for ONNX Runtime Training is not currently planned in the near future.

The planning of such work will mostly be determined by the need of such bindings for internal production scenarios.

Having said that, external contributions are welcome, and we are happy to engage with contributors to provide necessary support for such work.

vtantia commented 1 year ago

Thanks @baijumeswani for the detailed reply! Slightly unrelated to the issue but saw your recent blog post on ONNX Runtime Training as well and it was very explained so thanks for all your contributions.