fix bug in multi-batch_test (open-mmlab#985)

wjkim81 · Oct 25, 2021 · 2b4a5cf · 2b4a5cf
1 parent 59dc5e7
commit 2b4a5cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/test.py b/tools/test.py
@@ -130,8 +130,8 @@ def main():
         **loader_cfg,
         **dict(shuffle=False, drop_last=False),
         **dict(workers_per_gpu=cfg.data.get('workers_per_gpu', 1)),
-        **cfg.data.get('test_dataloader', {}),
-        **dict(samples_per_gpu=1)
+        **dict(samples_per_gpu=cfg.data.get('samples_per_gpu', 1)),
+        **cfg.data.get('test_dataloader', {})
     }
     data_loader = build_dataloader(dataset, **test_loader_cfg)