Tflite on-device training GPU

TensorFlow General Discussion
, ,
1

Following the official instructions, tflite can train models on a device. However, I was faced a lot of issues when trying to train a model using GPU on the device. Also in the TensorFlow blog, it was written that training on the device using the GPU is only in the plans. So I would like to know if it is possible to train models on the device using a GPU and tflite?

3

@Mykola_Kovbasa,

Welcome to the Tensorflow Forum!

We can build TensorFlow Lite using CMake with the GPU flag

-DTFLITE_ENABLE_GPU = True

For more details please refer here.

Thank you!