From 31a401bd629bb342ea602834e60d167470fc2dfa Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Fri, 23 Jun 2023 18:35:02 +0200 Subject: [PATCH] Update calibrate.py --- src/brevitas/graph/calibrate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas/graph/calibrate.py b/src/brevitas/graph/calibrate.py index 3fa85288f..f7a0dc0e1 100644 --- a/src/brevitas/graph/calibrate.py +++ b/src/brevitas/graph/calibrate.py @@ -138,7 +138,7 @@ def disable_act_quant_hook(self, module, inp, output): def disable_act_quantization(self, model, is_training, collect_stats=False): for module in model.modules(): if isinstance(module, ActQuantProxyFromInjector): - module.disable_quant = collect_stats + module.disable_quant = not collect_stats hook = module.register_forward_hook(self.disable_act_quant_hook) module.train(is_training) self.disable_act_quant_hooks.append(hook)