From a1d6fc95042c9df9dd07fe537a40f3bedc62ef72 Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Wed, 25 Mar 2020 18:27:17 -0400 Subject: [PATCH] [Tutorial][Quantization] Fix incorrect name of calibration mode (#5150) --- tutorials/frontend/deploy_quantized.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorials/frontend/deploy_quantized.py b/tutorials/frontend/deploy_quantized.py index 5af9fc950bc2..2586318321e2 100644 --- a/tutorials/frontend/deploy_quantized.py +++ b/tutorials/frontend/deploy_quantized.py @@ -132,7 +132,7 @@ def quantize(mod, params, data_aware): with relay.quantize.qconfig(calibrate_mode='kl_divergence', weight_scale='max'): mod = relay.quantize.quantize(mod, params, dataset=calibrate_dataset()) else: - with relay.quantize.qconfig(calibrate_mode='global', global_scale=8.0): + with relay.quantize.qconfig(calibrate_mode='global_scale', global_scale=8.0): mod = relay.quantize.quantize(mod, params) return mod