From 408f2954168ee836988567286e92dc6c88c5bfbb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20B=C3=A9dard-Couture?= Date: Sun, 5 May 2024 10:01:35 -0400 Subject: [PATCH] Remove unnecessary size check in TensorRT v10 block --- frigate/detectors/plugins/tensorrt.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index 0f41d7a8a..d9006fe86 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -175,9 +175,6 @@ class TensorRtDetector(DetectionApi): if len(binding_dims) == 4: # explicit batch case (TensorRT 7+) size = trt.volume(binding_dims) - elif len(binding_dims) == 3: - # implicit batch case (TensorRT 6 or older) - size = trt.volume(binding_dims) * self.engine.max_batch_size else: raise ValueError( "bad dims of binding %s: %s" % (binding, str(binding_dims))