Skip to content

Commit 0d3002a

Browse files
authored
Change samples_per_gpu in _infer_model(Detection) (#2041)
Change samples_per_gpu in _infer_model
1 parent ad0c960 commit 0d3002a

File tree

1 file changed

+1
-4
lines changed
  • otx/algorithms/detection/adapters/mmdet

1 file changed

+1
-4
lines changed

otx/algorithms/detection/adapters/mmdet/task.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -320,10 +320,7 @@ def _infer_model(
320320
cfg = self.configure(False, "test", None)
321321
logger.info("infer!")
322322

323-
samples_per_gpu = cfg.data.test_dataloader.get("samples_per_gpu", 1)
324-
if samples_per_gpu > 1:
325-
# Replace 'ImageToTensor' to 'DefaultFormatBundle'
326-
cfg.data.test.pipeline = replace_ImageToTensor(cfg.data.test.pipeline)
323+
samples_per_gpu = 1
327324

328325
# Data loader
329326
mm_dataset = build_dataset(cfg.data.test)

0 commit comments

Comments
 (0)