From c93aba2d344fc6f061d57f95788ce220d0e3715e Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 6 Jul 2021 16:33:33 +0800 Subject: [PATCH] fix slim export model error --- deploy/slim/quantization/export_model.py | 2 +- test/ocr_det_params.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py index 87d2f333..d94e5303 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 da7e034b..bdfd4d4f 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 -- GitLab