Skip to content

Commit

Permalink
[Tutorial][Quantization] Fix incorrect name of calibration mode (apac…
Browse files Browse the repository at this point in the history
  • Loading branch information
vinx13 authored and zhiics committed Apr 17, 2020
1 parent 06b75d1 commit ce9b6a5
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 ce9b6a5

Please sign in to comment.