diff --git a/ssod/datasets/builder.py b/ssod/datasets/builder.py index 91a57ab..1c02d54 100644 --- 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