Skip to content

Commit 4fb6dd4

Browse files
authored
Fix ONNX opset inconsistency with parseargs and run args (ultralytics#5937)
1 parent 5bdb28e commit 4fb6dd4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

export.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -325,7 +325,7 @@ def run(data=ROOT / 'data/coco128.yaml', # 'dataset.yaml path'
325325
int8=False, # CoreML/TF INT8 quantization
326326
dynamic=False, # ONNX/TF: dynamic axes
327327
simplify=False, # ONNX: simplify model
328-
opset=12, # ONNX: opset version
328+
opset=14, # ONNX: opset version
329329
verbose=False, # TensorRT: verbose log
330330
workspace=4, # TensorRT: workspace size (GB)
331331
topk_per_class=100, # TF.js NMS: topk per class to keep

0 commit comments

Comments
 (0)