TensorRT FP16 export with `dynamic=True` (#7870)

12 months ago · 9bdd784bcc
parent 51fb3d607e
commit 9bdd784bcc
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