如何在TFlite模型上进行微调

Question

我想根据自己的数据微调模型。但是，该模型以tflite格式分发。反正有没有从tflite文件中提取模型架构和参数？

Answer 1

我不确定您需要什么。但是，如果您想知道模型的确切架构，可以使用neutron进行查找。 您将获得如下内容：

为您提供信息，TensorFlow Lite并非经过微调。您需要微调经典的TensorFlow模型，然后将其转换为TensorFlow Lite。

Answer 2

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

使用tflite2onnx，类似于ONNX，然后导入到您选择的框架中。并非所有框架都可以从ONNX导入（例如PyTorch）。我相信您可以使用ONNXRuntime和MXNet进行培训。不确定是否可以使用TensorFlow进行训练。