fix det pact

2022-05-05 03:39:14 +00:00 · 2022-05-05 03:39:14 +00:00 · 74fdcfdb42
parent 5c7b6dc74b
commit 74fdcfdb42
1 changed files with 1 additions and 1 deletions
--- 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)