Unverified Commit a1d6fc95 by Wuwei Lin Committed by GitHub

[Tutorial][Quantization] Fix incorrect name of calibration mode (#5150)

parent 0c38b916
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment