From 2f22ff4ae66e86c53def72d52eb2e61ec924172e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 15 Feb 2024 12:41:12 +0100 Subject: [PATCH] Remove OpenVINO INT8 RTDETRDecoder ops (#8219) --- 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 2da51ba4..a8a58b27 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -455,7 +455,7 @@ class Exporter: LOGGER.warning(f"{prefix} WARNING ⚠️ >300 images recommended for INT8 calibration, found {n} images.") quantization_dataset = nncf.Dataset(dataset, transform_fn) ignored_scope = None - if isinstance(self.model.model[-1], (Detect, RTDETRDecoder)): + if isinstance(self.model.model[-1], Detect): # Includes all Detect subclasses like Segment, Pose, OBB, WorldDetect head_module_name = ".".join(list(self.model.named_modules())[-1][0].split(".")[:2])