diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py index 87d2f333c1923b088d32e5fb420ae7a5021d6b08..d94e53034a2bf67b364e6d91f83acfb9e5445b8a 100755 --- a/deploy/slim/quantization/export_model.py +++ b/deploy/slim/quantization/export_model.py @@ -101,7 +101,7 @@ def main(): quanter = QAT(config=quant_config) quanter.quantize(model) - init_model(config, model, logger) + init_model(config, model) model.eval() # build metric diff --git a/test/ocr_det_params.txt b/test/ocr_det_params.txt index da7e034bdd1ef1f593020d2dd5e809b472651b54..bdfd4d4f47431bca97437963e1dc56d1b57838bb 100644 --- a/test/ocr_det_params.txt +++ b/test/ocr_det_params.txt @@ -17,7 +17,7 @@ distill_train:null eval:tools/eval.py -c configs/det/det_mv3_db.yml -o Global.save_inference_dir:./output/ -Global.checkpoints: +Global.pretrained_model: norm_export:tools/export_model.py -c configs/det/det_mv3_db.yml -o quant_export:deploy/slim/quantization/export_model.py -c configs/det/det_mv3_db.yml -o fpgm_export:deploy/slim/prune/export_prune_model.py