mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-06-03 21:53:39 +08:00
fix det pact
This commit is contained in:
parent
5c7b6dc74b
commit
74fdcfdb42
@ -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)
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user