microsoft / onnxruntime-training-examples

Examples for using ONNX Runtime for model training.
MIT License
301 stars 60 forks source link

Learning on the edge android app example #101

Closed baijumeswani closed 1 year ago

baijumeswani commented 1 year ago

This example shows how to perform learning on the edge using onnxruntime training api in python and C++.

The android app performs the task of image classification using transfer learning (only the last layer is trainable). User can bring their own data to train the model and run inference on their custom classes.