如何对TFlite模型进行微调

Question

我想根据自己的数据对模型进行微调。但是，该模型是通过tflite格式分发的。有没有办法从tflite文件中提取模型架构和参数？

Answer 1

我不太清楚你需要什么。但是如果你想知道你的模型的确切架构，你可以使用neutron来找出答案。您将得到类似于以下内容的内容：

​

供您参考的是，TensorFlow精简版并不是用来优化的。您需要微调一个经典的TensorFlow模型，然后将其转换为TensorFlow精简版。

Answer 2

一种方法是将TFLite文件转换为另一种格式，并导入支持训练的深度学习框架。

例如ONNX，使用tflite2onnx，然后导入到您选择的框架中。并不是所有的框架都可以从ONNX导入(例如PyTorch)。我相信你可以和ONNXRuntime和MXNet一起训练。不确定是否可以使用TensorFlow进行训练。