TensorRT FP16 export with dynamic=True (#7870)

2024-01-29 21:54:06 +09:00 · 2024-01-29 21:54:06 +09:00 · 9bdd784bcc
commit 9bdd784bcc
parent 51fb3d607e
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@ -227,7 +227,7 @@ class Exporter:
        y = None
        for _ in range(2):
            y = model(im)  # dry runs
-        if self.args.half and (engine or onnx) and self.device.type != "cpu":
+        if self.args.half and onnx and self.device.type != "cpu":
            im, model = im.half(), model.half()  # to FP16

        # Filter warnings