From b273cf279ea0a3558f47d64a5433dc32b349ae9b Mon Sep 17 00:00:00 2001 From: shangliang Xu Date: Wed, 11 Aug 2021 15:05:29 +0800 Subject: [PATCH] add merge_config (#3938) --- tools/export_model.py | 2 ++ tools/train.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/tools/export_model.py b/tools/export_model.py index 3420cf27f..ddb7deb17 100644 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -103,6 +103,8 @@ def main(): if FLAGS.slim_config: cfg = build_slim_model(cfg, FLAGS.slim_config, mode='test') + # FIXME: Temporarily solve the priority problem of FLAGS.opt + merge_config(FLAGS.opt) check_config(cfg) check_gpu(cfg.use_gpu) check_version() diff --git a/tools/train.py b/tools/train.py index 55aed43ee..a49885cb4 100755 --- a/tools/train.py +++ b/tools/train.py @@ -128,6 +128,8 @@ def main(): if FLAGS.slim_config: cfg = build_slim_model(cfg, FLAGS.slim_config) + # FIXME: Temporarily solve the priority problem of FLAGS.opt + merge_config(FLAGS.opt) check.check_config(cfg) check.check_gpu(cfg.use_gpu) check.check_version() -- GitLab