【问题标题】:ValueError: Unknown layer: QuantizeLayerValueError:未知层:QuantizeLayer
【发布时间】: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 tensorflow2.0


    【解决方案1】:

    以下代码解决的问题:

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

    【讨论】:

      猜你喜欢
      • 2018-11-23
      • 2021-02-22
      • 2020-11-14
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2020-02-24
      相关资源
      最近更新 更多