如何量化预训练模型（例如 MobileNet）

Question

我正在使用 Tensorflow lite 框架来创建实验的量化模型。我想在我的 Raspberry Pi 上部署这个模型，但似乎使用预训练模型进行量化不起作用。

我尝试在 Colab 和 Kaggle 笔记本上执行以下代码。

from tensorflow.keras.applications import MobileNet

pretrained_model = MobileNet(include_top=False, weights='imagenet',input_shape=(32,32,3))

import tensorflow_model_optimization as tfmot

pretrained_model.trainable = True

q_pretrained_model = tfmot.quantization.keras.quantize_
model(pretrained_model)

它会抛出一个错误：-

to_quantize

只能是 keras 顺序模型或函数模型。

Answer 1

我面临保存问题。你找到解决办法了吗？

谢谢， 埃迪33