make opset_version optional

inisis · Jul 28, 2022 · fba0fd4 · fba0fd4
1 parent ffa06f0
commit fba0fd4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bin/fx/pytorch2caffe.py b/bin/fx/pytorch2caffe.py
@@ -14,7 +14,7 @@
 from converter.pytorch.fx.pytorch_caffe_parser import PytorchCaffeParser  # noqa
 
 class Runner(object):
-    def __init__(self, name, model, shape, opset_version, fuse=False, concrete_args=None):
+    def __init__(self, name, model, shape, opset_version=13, fuse=False, concrete_args=None):
         self.name = name
         self.model = model
         self.shape = shape

diff --git a/bin/fx/pytorch2onnx.py b/bin/fx/pytorch2onnx.py
@@ -13,7 +13,7 @@
 from converter.pytorch.fx.pytorch_onnx_parser import PytorchOnnxParser  # noqa
 
 class Runner(object):
-    def __init__(self, name, model, shape, opset_version, fuse=False):
+    def __init__(self, name, model, shape, opset_version=13, fuse=False):
         self.name = name
         self.model = model
         self.shape = shape