Skip to content

Commit

Permalink
[Tutorial][Quantization] Fix incorrect name of calibration mode (#5150)
Browse files Browse the repository at this point in the history
  • Loading branch information
vinx13 authored Mar 25, 2020
1 parent 0c38b91 commit a1d6fc9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tutorials/frontend/deploy_quantized.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit a1d6fc9

Please sign in to comment.