Channingss 5 years ago
parent
commit
d83371a252
1 changed files with 2 additions and 1 deletions
  1. 2 1
      paddlex/cv/models/base.py

+ 2 - 1
paddlex/cv/models/base.py

@@ -408,7 +408,6 @@ class BaseAPI:
                 for v in test_outputs_names
             ]
             # Make graph
-            onnx_name = 'test'
             onnx_graph = helper.make_graph(
                 nodes=onnx_nodes,
                 name=onnx_name,
@@ -431,6 +430,8 @@ class BaseAPI:
             if onnx_model is not None:
                 try:
                     onnx_model_file = osp.join(save_dir, onnx_name)
+                    if not os.path.exists(save_dir):
+                        os.mkdir(save_dir)
                     with open(onnx_model_file, 'wb') as f:
                         f.write(onnx_model.SerializeToString())
                     print(