Skip to content

Commit

Permalink
Update on "[reland][quant][eagermode] Move custom_module registration…
Browse files Browse the repository at this point in the history
… to prepare/convert_custom_config_dict (#46293)"

Summary:

Test Plan: Imported from OSS

Reviewed By: raghuramank100

Differential Revision: [D24322747](https://our.internmc.facebook.com/intern/diff/D24322747)

[ghstack-poisoned]
  • Loading branch information
jerryzh168 committed Oct 19, 2020
1 parent 5a2434b commit a930903
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/quantization/quantize.py
Expand Up @@ -215,7 +215,7 @@ def prepare(model, inplace=False, allow_list=None,
qconfig_propagation_list = allow_list
if qconfig_propagation_list is None:
qconfig_propagation_list = get_qconfig_propagation_list()
propagate_qconfig_(model, qconfig_dict=None)
propagate_qconfig_(model, qconfig_dict=None)

# sanity check common API misusage
if not any(hasattr(m, 'qconfig') and m.qconfig for m in model.modules()):
Expand Down

0 comments on commit a930903

Please sign in to comment.