CODEWITHSUNDEEP
tensorflow-lite
N.Lee
N.Lee

Reputation: 1036

Does tensorflow lite support on device training at now

Because Google issued several paper on federated learning on mobile device, like Gboard, but I can not find docs on training on TFLite website, does TFLite support on device training at now?

Upvotes: 7

Views: 2245

Answers (1)

aselle
aselle

Reputation: 639

We do not support general training (or federated training) on TensorFlow Lite. We however have released a tutorial on doing on-device personalization with TF lite. https://blog.tensorflow.org/2019/12/example-on-device-model-personalization.html

It basically goes through the process of exporting subgraphs that do training substeps and evaluation to stage training efforts manually. Personalization is the most practical on-device training mechanism right now, so hopefully this addresses your needed.

We are also working on a more comprehensive training solution that will eventually support things like federated learning and also more easy-to-use workflows.

Upvotes: 6

Related Questions