Representative dataset for QAT

If I am preparing my model quantisation aware using tfmot.quantization.keras.quantize_model, do I need a representative dataset? Or is it only needed for Post-Training quant?

@Chinya07,

Welcome to the Tensorflow Forum!

Sorry for the delay in responding.

No. Only Post Training Quantization requires a representative dataset because it computes scale after the network has been trained.

Thank you.

1 Like