How to apply quantization aware training on EfficientDet keras model?

google / automl

Google Brain AutoML

Apache License 2.0

6.18k stars 1.45k forks source link

How to apply quantization aware training on EfficientDet keras model? #1189

Open kimwj94 opened 1 year ago

kimwj94 commented 1 year ago

I'm trying to do quantization aware training on EfficientDet keras model. I tried use Nvidia TensorFlow-2.x-Quantization-Toolkit, but I got error: ValueError: to_quantize can only either be a tf.keras Sequential or Functional model.

How can I modify the model to do QAT?