Tflite on-device training GPU

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?



Welcome to the Tensorflow Forum!

We can build TensorFlow Lite using CMake with the GPU flag


For more details please refer here.

Thank you!