glenn-jocher
commited on
Commit
•
8b18b66
1
Parent(s):
8ac96b7
ONNX opset 13 (#4566)
Browse files
export.py
CHANGED
@@ -176,7 +176,7 @@ def parse_opt():
|
|
176 |
parser.add_argument('--optimize', action='store_true', help='TorchScript: optimize for mobile')
|
177 |
parser.add_argument('--dynamic', action='store_true', help='ONNX: dynamic axes')
|
178 |
parser.add_argument('--simplify', action='store_true', help='ONNX: simplify model')
|
179 |
-
parser.add_argument('--opset', type=int, default=
|
180 |
opt = parser.parse_args()
|
181 |
return opt
|
182 |
|
|
|
176 |
parser.add_argument('--optimize', action='store_true', help='TorchScript: optimize for mobile')
|
177 |
parser.add_argument('--dynamic', action='store_true', help='ONNX: dynamic axes')
|
178 |
parser.add_argument('--simplify', action='store_true', help='ONNX: simplify model')
|
179 |
+
parser.add_argument('--opset', type=int, default=13, help='ONNX: opset version')
|
180 |
opt = parser.parse_args()
|
181 |
return opt
|
182 |
|