From 4fb6dd4b26a4d1c39a1e2565999be62f53fb7c71 Mon Sep 17 00:00:00 2001 From: Diego Montes <54745152+d57montes@users.noreply.github.com> Date: Thu, 9 Dec 2021 17:10:16 -0500 Subject: [PATCH] Fix ONNX opset inconsistency with parseargs and run args (#5937) --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 88d03a2..4f83c75 100644 --- a/export.py +++ b/export.py @@ -325,7 +325,7 @@ def run(data=ROOT / 'data/coco128.yaml', # 'dataset.yaml path' int8=False, # CoreML/TF INT8 quantization dynamic=False, # ONNX/TF: dynamic axes simplify=False, # ONNX: simplify model - opset=12, # ONNX: opset version + opset=14, # ONNX: opset version verbose=False, # TensorRT: verbose log workspace=4, # TensorRT: workspace size (GB) topk_per_class=100, # TF.js NMS: topk per class to keep