ValueError:未知层:QuantizeLayer

Rya*_*Liu 2 tensorflow tensorflow2.0

使用tensorflow_model_optimization进行QAT训练后,遵循教程

然后使用Keras加载量化模型将显示未知层:QuantizeLayer

   tf.keras.models.load_model(keras_quantized_model_file)
Run Code Online (Sandbox Code Playgroud)

Rya*_*Liu 11

问题通过如下代码解决:

import tensorflow_model_optimization as tfmot

with tfmot.quantization.keras.quantize_scope():
    loaded_model = tf.keras.models.load_model(keras_quantized_model_file)
Run Code Online (Sandbox Code Playgroud)