如何将TFLite模型转换为量化的TFLite模型?

时间:2020-10-15 13:08:50

标签: tensorflow model converters quantization

我有一个tflite文件,我想对其进行量化。

如何将TFLite模型转换为量化的TFLite模型?

1 个答案:

答案 0 :(得分:1)

根据您使用的是keras还是tfhub,您可以简单地执行以下操作(这是假设TF模型不支持量化):

converter = tf.lite.TFLiteConverter.from_keras_model(non_qat_model)
converter.optimizations = [tf.lite.Optimize.DEFAULT]
quantized_tflite_model = converter.convert()

您可以在此处进一步参考链接: https://www.tensorflow.org/lite/performance/post_training_quantization