Change samples_per_gpu in _infer_model(Detection) (#2041)

Change samples_per_gpu in _infer_model
openvinotoolkit · Apr 20, 2023 · 0d3002a · 0d3002a
1 parent ad0c960
commit 0d3002a
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/otx/algorithms/detection/adapters/mmdet/task.py b/otx/algorithms/detection/adapters/mmdet/task.py
@@ -320,10 +320,7 @@ def _infer_model(
         cfg = self.configure(False, "test", None)
         logger.info("infer!")
 
-        samples_per_gpu = cfg.data.test_dataloader.get("samples_per_gpu", 1)
-        if samples_per_gpu > 1:
-            # Replace 'ImageToTensor' to 'DefaultFormatBundle'
-            cfg.data.test.pipeline = replace_ImageToTensor(cfg.data.test.pipeline)
+        samples_per_gpu = 1
 
         # Data loader
         mm_dataset = build_dataset(cfg.data.test)