ValueError:未知层:QuantizeLayer

时间:2020-09-30 05:29:21

标签: tensorflow tensorflow2.0

在使用tensorflow_model_optimization进行QAT培训之后,遵循tutorial

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

   tf.keras.models.load_model(keras_quantized_model_file)

1 个答案:

答案 0 :(得分:1)

该问题由以下代码解决:

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