Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Fix doc
Browse files Browse the repository at this point in the history
  • Loading branch information
reminisce committed Mar 25, 2018
1 parent 21512de commit 7be4936
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions python/mxnet/contrib/quantization.py
Original file line number Diff line number Diff line change
Expand Up @@ -452,9 +452,11 @@ def quantize_model(sym, arg_params, aux_params,
otherwise, no information of the layer's output will be collected. If not provided,
all the layers' outputs that need requantization will be collected.
logger : Object
A logging object for printing information during the process of quantization.
Returns
`(qsym, qarg_params, aux_params)`
-------
tuple
A tuple of quantized symbol, quantized arg_params, and aux_params.
-------
"""
Expand All @@ -471,7 +473,8 @@ def quantize_model(sym, arg_params, aux_params,
idx = nodes.list_outputs().index(sym_name + '_output')
excluded_syms.append(nodes[idx])
logger.info('Quantizing symbol')
qsym = _quantize_symbol(sym, excluded_symbols=excluded_syms, offline_params=arg_params.keys())
qsym = _quantize_symbol(sym, excluded_symbols=excluded_syms,
offline_params=list(arg_params.keys()))

logger.info('Quantizing parameters')
qarg_params = _quantize_params(qsym, arg_params)
Expand Down

0 comments on commit 7be4936

Please sign in to comment.