From 2940110974dcc06a861e058021788ae372ba2c50 Mon Sep 17 00:00:00 2001 From: Burhan <62214284+Burhan-Q@users.noreply.github.com> Date: Fri, 10 May 2024 11:55:07 -0400 Subject: [PATCH] Enforce `dynamic=True` for TensorRT and OpenVINO INT8 (#11788) --- ultralytics/engine/exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index fa681099..7e1baa46 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -200,7 +200,7 @@ class Exporter: self.args.half = False assert not self.args.dynamic, "half=True not compatible with dynamic=True, i.e. use only one." self.imgsz = check_imgsz(self.args.imgsz, stride=model.stride, min_dim=2) # check image size - if self.args.int8 and engine: + if self.args.int8 and (engine or xml): self.args.dynamic = True # enforce dynamic to export TensorRT INT8; ensures ONNX is dynamic if self.args.optimize: assert not ncnn, "optimize=True not compatible with format='ncnn', i.e. use optimize=False"