|
|
@ -83,7 +83,7 @@ int8: False # (bool) CoreML/TF INT8 quantization |
|
|
|
dynamic: False # (bool) ONNX/TF/TensorRT: dynamic axes |
|
|
|
dynamic: False # (bool) ONNX/TF/TensorRT: dynamic axes |
|
|
|
simplify: True # (bool) ONNX: simplify model using `onnxslim` |
|
|
|
simplify: True # (bool) ONNX: simplify model using `onnxslim` |
|
|
|
opset: # (int, optional) ONNX: opset version |
|
|
|
opset: # (int, optional) ONNX: opset version |
|
|
|
workspace: 4 # (int) TensorRT: workspace size (GB) |
|
|
|
workspace: None # (float, optional) TensorRT: workspace size (GiB), `None` will let TensorRT auto-allocate memory |
|
|
|
nms: False # (bool) CoreML: add NMS |
|
|
|
nms: False # (bool) CoreML: add NMS |
|
|
|
|
|
|
|
|
|
|
|
# Hyperparameters ------------------------------------------------------------------------------------------------------ |
|
|
|
# Hyperparameters ------------------------------------------------------------------------------------------------------ |
|
|
|