From c7e819c9ec6c7e0577d3546c42b012a9f9f778c1 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Fri, 25 Oct 2024 07:44:04 +0000 Subject: [PATCH] Auto-format by https://ultralytics.com/actions --- export.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/export.py b/export.py index 25acf9152..f5dc0684e 100644 --- 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.