ValueError: Unknown layer: QuantizeLayer

user_2469488 0
Viewed 902

After QAT training with tensorflow_model_optimization followed the tutorial

then use Keras to load the quantized model will show Unknown layer: QuantizeLayer

   tf.keras.models.load_model(keras_quantized_model_file)
tensorflow2.0 tensorflow
1 Answers

The problem solved by the following code:

import tensorflow_model_optimization as tfmot

with tfmot.quantization.keras.quantize_scope():
    loaded_model = tf.keras.models.load_model(keras_quantized_model_file)
user_24694880
Related