Ver Fonte

fix bug for request paddle2onnx.program2onnx

Channingss há 5 anos atrás
pai
commit
f571beed70
1 ficheiros alterados com 1 adições e 1 exclusões
  1. 1 1
      paddlex/converter.py

+ 1 - 1
paddlex/converter.py

@@ -287,6 +287,6 @@ def export_onnx_model(model, save_dir, opset_version=10):
 
     p2o.program2onnx(
         model.test_prog,
-        save_dir,
         scope=model.scope,
+        save_file=save_dir,
         opset_version=opset_version)