bug fix for init sampler during testing

3ef19ea7 · Kai Chen · 5f44ba8a · 3ef19ea7
Commit 3ef19ea7 authored Oct 25, 2018 by Kai Chen
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 3 deletions

mmdet/datasets/loader/build_loader.py mmdet/datasets/loader/build_loader.py +4 -3

No files found.
--- a/mmdet/datasets/loader/build_loader.py
+++ b/mmdet/datasets/loader/build_loader.py
@@ -25,12 +25,13 @@ def build_dataloader(dataset,
        batch_size = imgs_per_gpu
        num_workers = workers_per_gpu
    else:
-        sampler = GroupSampler(dataset, imgs_per_gpu)
+        if not kwargs.get('shuffle', True):
+            sampler = None
+        else:
+            sampler = GroupSampler(dataset, imgs_per_gpu)
        batch_size = num_gpus * imgs_per_gpu
        num_workers = num_gpus * workers_per_gpu
-    if not kwargs.get('shuffle', True):
-        sampler = None
    data_loader = DataLoader(
        dataset,