如何在TFlite模型上进行微调

问题描述

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

解决方法

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

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

,

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

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