From 18e22211050f1b5089dc335dc618423af451e27a Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Wed, 25 Mar 2020 15:49:08 -0400 Subject: [PATCH] [Tutorial][Quantization] Fix incorrect name of calibration mode --- 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