Auto-format by https://ultralytics.com/actions

2024-10-25 07:44:04 +00:00 · 2024-10-25 07:44:04 +00:00 · c7e819c9ec
parent 10bf52d087
commit c7e819c9ec
1 changed files with 3 additions and 2 deletions
--- a/export.py
+++ b/export.py
@ -593,8 +593,9 @@ def export_coreml(model, im, file, int8, half, nms, mlmodel, prefix=colorstr("Co


@try_export
-def export_engine(model, im, file, half, dynamic, simplify, workspace=4, verbose=False, cache="",
-                  prefix=colorstr("TensorRT:")):
+def export_engine(
+    model, im, file, half, dynamic, simplify, workspace=4, verbose=False, cache="", prefix=colorstr("TensorRT:")
+):
    """
    Export a YOLOv5 model to TensorRT engine format, requiring GPU and TensorRT>=7.0.0.