title: Tensorflow Quantization Aware Training
body: <p>I want to quantize a DenseNet model. I am using Tensorflow 2.4.</p>
  <pre><code>import tensorflow_model_optimization as tfmot
  model = tf.keras.applications.DenseNet121(include_top=True,weights=None,input_tensor=None,input_shape=None,pooling=None,classes=1000) 
  quantize_model = tfmot.quantization.keras.quantize_model
  model = quantize_model(model)
  </code></pre>
  <p>But I got the following message:</p>
  <p>RuntimeError: Layer conv2_block1_0_bn:&lt;class 'tensorflow.python.keras.layers.normalization_v2.BatchNormalization'&gt; is not supported. You can quantize this layer by passing a <code>tfmot.quantization.keras.QuantizeConfig</code> instance to the <code>quantize_annotate_layer</code> API.</p>
  <p>Is there a way how I can do this.
  I can not change the keras code.</p>