在使用tensorflow_model_optimization进行QAT培训之后,遵循tutorial
然后使用Keras加载量化模型将显示未知层:QuantizeLayer
tf.keras.models.load_model(keras_quantized_model_file)
答案 0 :(得分:1)
该问题由以下代码解决:
with tfmot.quantization.keras.quantize_scope():
loaded_model = tf.keras.models.load_model(keras_quantized_model_file)