microsoft / onnxruntime-training-examples

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

Add ios app example for on device training #158

Closed vraspar closed 1 year ago

vraspar commented 1 year ago

Description

Add iOS app example to showcase new Objective-C/Swift training API. The sample app performs on device speaker verification by using transfers learning on wav2vec model. It shows how new training APIs can be used on iOS platform for personalization and fine tuning tasks.

The Trainer.swift file contains the main training loop and it is responsible for training the model.