From 74fdcfdb42d82ead0fd72b06dd68aae2dac074be Mon Sep 17 00:00:00 2001 From: andyjpaddle Date: Thu, 5 May 2022 03:39:14 +0000 Subject: [PATCH] fix det pact --- deploy/slim/quantization/quant.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy/slim/quantization/quant.py b/deploy/slim/quantization/quant.py index a44a8b58..64521b5e 100755 --- a/deploy/slim/quantization/quant.py +++ b/deploy/slim/quantization/quant.py @@ -166,7 +166,7 @@ def main(config, device, logger, vdl_writer): for key in config['Architecture']["Models"]: freeze_params = freeze_params or config['Architecture']['Models'][ key].get('freeze_params', False) - act = None if freeze_params else 'PACT' + act = None if freeze_params else PACT quanter = QAT(config=quant_config, act_preprocess=act) quanter.quantize(model) -- GitLab