From f5de2fec70060ffb1cc29273c60d5f8cf4d465db Mon Sep 17 00:00:00 2001
From: Jason <jjj2070@163.com>
Date: Fri, 17 Jul 2020 22:34:22 +0800
Subject: [PATCH] Update convert.py

---
 x2paddle/convert.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/x2paddle/convert.py b/x2paddle/convert.py
index d3503ef..fe0b09e 100644
--- a/x2paddle/convert.py
+++ b/x2paddle/convert.py
@@ -192,12 +192,12 @@ def onnx2paddle(model_path, save_dir, params_merge=False):
     print("Paddle model and code generated.")
 
 
-def paddle2onnx(model_path, save_dir, opset_number):
+def paddle2onnx(model_path, save_dir, opset_version=10):
     from x2paddle.decoder.paddle_decoder import PaddleDecoder
     from x2paddle.op_mapper.paddle2onnx.paddle_op_mapper import PaddleOpMapper
     model = PaddleDecoder(model_path, '__model__', '__params__')
     mapper = PaddleOpMapper()
-    mapper.convert(model.program, save_dir, opset_number=opset_number)
+    mapper.convert(model.program, save_dir, opset_number=opset_version)
 
 
 def main():
-- 
GitLab