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)
| 归档时间: |
|
| 查看次数: |
1481 次 |
| 最近记录: |