rename inference model save path

0ff2aef2 · 文幕地方 · 4fd696cc · 0ff2aef2
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

tools/export_model.py tools/export_model.py +1 -2

未找到文件。
--- 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(