Diego Montes
commited on
Commit
•
4fb6dd4
1
Parent(s):
5bdb28e
Fix ONNX opset inconsistency with parseargs and run args (#5937)
Browse files
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=
|
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
|