rename inference model save path

4 years ago · 0ff2aef299
parent 4fd696ccdf
commit 0ff2aef299
1 changed files with 1 additions and 2 deletions
--- a/tools/export_model.py
+++ b/tools/export_model.py
@ -57,8 +57,7 @@ def main():
    init_model(config, model, logger)
    model.eval()

-    save_path = '{}/{}/inference'.format(FLAGS.output_path,
-                                         config['Architecture']['model_type'])
+    save_path = '{}/inference'.format(FLAGS.output_path)
    infer_shape = [3, 32, 100] if config['Architecture'][
        'model_type'] != "det" else [3, 640, 640]
    model = to_static(