From 9abf3fb0d2d6cb976ddb66987972866b03d7fba4 Mon Sep 17 00:00:00 2001 From: WenmuZhou <572459439@qq.com> Date: Mon, 10 Jan 2022 02:30:26 +0000 Subject: [PATCH] fix merge config error --- deploy/slim/quantization/export_model.py | 2 +- tools/export_model.py | 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 0cb86108..34cf80f5 100755 --- a/deploy/slim/quantization/export_model.py +++ b/deploy/slim/quantization/export_model.py @@ -76,7 +76,7 @@ def main(): } FLAGS = ArgsParser().parse_args() config = load_config(FLAGS.config) - merge_config(FLAGS.opt) + config = merge_config(config, FLAGS.opt) logger = get_logger() # build post process diff --git a/tools/export_model.py b/tools/export_model.py index 9ed8e1b6..695af5c8 100755 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -85,7 +85,7 @@ def export_single_model(model, arch_config, save_path, logger): def main(): FLAGS = ArgsParser().parse_args() config = load_config(FLAGS.config) - merge_config(FLAGS.opt) + config = merge_config(config, FLAGS.opt) logger = get_logger() # build post process -- GitLab