glenn-jocher commited on
Commit
8b18b66
1 Parent(s): 8ac96b7

ONNX opset 13 (#4566)

Browse files
Files changed (1) hide show
  1. export.py +1 -1
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=12, help='ONNX: opset version')
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