diff --git a/deploy/slim/quantization/export_model.py b/deploy/slim/quantization/export_model.py index 0cb86108d2275dc6ee1a74e118c27b94131975d3..34cf80f5e5566707a08d15ddeaaa51348dcd9acf 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 9ed8e1b6ace89ded030c946870551c8e078d7340..695af5c8bd092ec9a0ef806f8170cc686b194b73 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