Diego Montes commited on
Commit
4fb6dd4
1 Parent(s): 5bdb28e

Fix ONNX opset inconsistency with parseargs and run args (#5937)

Browse files
Files changed (1) hide show
  1. export.py +1 -1
export.py CHANGED
@@ -325,7 +325,7 @@ def run(data=ROOT / 'data/coco128.yaml', # 'dataset.yaml path'
325
  int8=False, # CoreML/TF INT8 quantization
326
  dynamic=False, # ONNX/TF: dynamic axes
327
  simplify=False, # ONNX: simplify model
328
- opset=12, # ONNX: opset version
329
  verbose=False, # TensorRT: verbose log
330
  workspace=4, # TensorRT: workspace size (GB)
331
  topk_per_class=100, # TF.js NMS: topk per class to keep
 
325
  int8=False, # CoreML/TF INT8 quantization
326
  dynamic=False, # ONNX/TF: dynamic axes
327
  simplify=False, # ONNX: simplify model
328
+ opset=14, # ONNX: opset version
329
  verbose=False, # TensorRT: verbose log
330
  workspace=4, # TensorRT: workspace size (GB)
331
  topk_per_class=100, # TF.js NMS: topk per class to keep