From 0ff2aef299b319edc2b783246e0b42e45a23b890 Mon Sep 17 00:00:00 2001
From: WenmuZhou <572459439@qq.com>
Date: Wed, 9 Dec 2020 23:59:29 +0800
Subject: [PATCH] rename inference model save path

---
 tools/export_model.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tools/export_model.py b/tools/export_model.py
index 46a8a8b8..b6c03efb 100755
--- 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(
-- 
GitLab