MMZ
MMZ

Reputation: 21

How to do fine tuning on TFlite model

I would like to fine tune a model on my own data. However the model is distributed by tflite format. Is there anyway to extract the model architecture and parameters out of the tflite file?

Upvotes: 2

Views: 1596

Answers (2)

T.K
T.K

Reputation: 464

I'm not sure to understand what you need. But if you want to know the exact architecture of your model you can use neutron to find out. You will get something like the this : enter image description here

And for your information TensorFlow Lite is not meant to be finetuned. You need to finetune a classic TensorFlow model and then convert it to TensorFlow Lite.

Upvotes: 0

Prunus Persica
Prunus Persica

Reputation: 1203

One approach could be to convert the TFLite file to another format, and import into a deep learning framework that supports training.

Something like ONNX, using tflite2onnx, and then import into a framework of your choice. Not all frameworks can import from ONNX (e.g. PyTorch). I believe you can train with ONNXRuntime, and MXNet. Unsure if you can train using TensorFlow.

Upvotes: 1

Related Questions