Skip to content

GitLab

Commit 74fdcfdb authored by andyjpaddle's avatar andyjpaddle
Browse files

fix det pact

parent 5c7b6dc7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
deploy/slim/quantization/quant.py
View file @ 74fdcfdb
...@@ -166,7 +166,7 @@ def main(config, device, logger, vdl_writer): ...@@ -166,7 +166,7 @@ def main(config, device, logger, vdl_writer):
for key in config['Architecture']["Models"]: for key in config['Architecture']["Models"]:
freeze_params = freeze_params or config['Architecture']['Models'][ freeze_params = freeze_params or config['Architecture']['Models'][
key].get('freeze_params', False) 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 = QAT(config=quant_config, act_preprocess=act)
quanter.quantize(model) quanter.quantize(model)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment