From fbbaf5312db4fa3c1cb87980b726ba50a421d5fa Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Mon, 5 Jul 2021 03:31:32 +0000 Subject: [PATCH] fix quant export --- deploy/slim/quantization/export_model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py index 100b107a..942f59e0 100755 --- a/deploy/slim/quantization/export_model.py +++ b/deploy/slim/quantization/export_model.py @@ -92,9 +92,11 @@ def main(): # build dataloader valid_dataloader = build_dataloader(config, 'Eval', device, logger) + use_srn = config['Architecture']['algorithm'] == "SRN" + model_type = config['Architecture']['model_type'] # start eval metirc = program.eval(model, valid_dataloader, post_process_class, - eval_class) + eval_class, model_type, use_srn) logger.info('metric eval ***************') for k, v in metirc.items(): logger.info('{}:{}'.format(k, v)) -- GitLab