From 4597879f5a4f3f0690b3af54f507d17fa0c977c1 Mon Sep 17 00:00:00 2001 From: Diego Montes Date: Thu, 9 Dec 2021 13:40:05 -0500 Subject: [PATCH] fix small inconsistency with parseargs and run args --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 88d03a2c9475..4f83c75c89a0 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