From a9309038abbd77804609ceb11d77292c648b699f Mon Sep 17 00:00:00 2001 From: Jerry Zhang Date: Mon, 19 Oct 2020 11:26:18 -0700 Subject: [PATCH] Update on "[reland][quant][eagermode] Move custom_module registration 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] --- torch/quantization/quantize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch/quantization/quantize.py b/torch/quantization/quantize.py index 8f27e7281154..f50d32031e08 100644 --- a/torch/quantization/quantize.py +++ b/torch/quantization/quantize.py @@ -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()):