fix sampler bug

microsoft · Oct 13, 2021 · e72c59c · e72c59c
1 parent 20e2618
commit e72c59c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ssod/datasets/builder.py b/ssod/datasets/builder.py
@@ -64,7 +64,11 @@ def build_dataloader(
         batch_size = samples_per_gpu
         num_workers = workers_per_gpu
     else:
-        sampler = build_sampler(sampler_cfg, default_sampler_cfg) if shuffle else None
+        sampler = (
+            build_sampler(sampler_cfg, default_args=default_sampler_cfg)
+            if shuffle
+            else None
+        )
         batch_size = num_gpus * samples_per_gpu
         num_workers = num_gpus * workers_per_gpu