diff --git a/tools/export_model.py b/tools/export_model.py index 3420cf27f810c7f325e0b35a31b790f594803749..ddb7deb177ca4bd16fe1a4a95887813b3858b573 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 55aed43ee032cde13f66cb2288fb3e412c9f70db..a49885cb4f839419b32c0708124005349cf29737 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()