【发布时间】:2022-02-04 15:06:14
【问题描述】:
在使用 tensorflow_model_optimization 进行 QAT 训练后,遵循tutorial
然后使用Keras加载量化后的模型会显示Unknown layer: QuantizeLayer
tf.keras.models.load_model(keras_quantized_model_file)
【问题讨论】:
在使用 tensorflow_model_optimization 进行 QAT 训练后,遵循tutorial
然后使用Keras加载量化后的模型会显示Unknown layer: QuantizeLayer
tf.keras.models.load_model(keras_quantized_model_file)
【问题讨论】:
以下代码解决的问题:
import tensorflow_model_optimization as tfmot
with tfmot.quantization.keras.quantize_scope():
loaded_model = tf.keras.models.load_model(keras_quantized_model_file)
【讨论】: