Commit 3ef19ea7 authored by Kai Chen's avatar Kai Chen
Browse files

bug fix for init sampler during testing

parent 5f44ba8a
...@@ -24,13 +24,14 @@ def build_dataloader(dataset, ...@@ -24,13 +24,14 @@ def build_dataloader(dataset,
rank) rank)
batch_size = imgs_per_gpu batch_size = imgs_per_gpu
num_workers = workers_per_gpu num_workers = workers_per_gpu
else:
if not kwargs.get('shuffle', True):
sampler = None
else: else:
sampler = GroupSampler(dataset, imgs_per_gpu) sampler = GroupSampler(dataset, imgs_per_gpu)
batch_size = num_gpus * imgs_per_gpu batch_size = num_gpus * imgs_per_gpu
num_workers = num_gpus * workers_per_gpu num_workers = num_gpus * workers_per_gpu
if not kwargs.get('shuffle', True):
sampler = None
data_loader = DataLoader( data_loader = DataLoader(
dataset, dataset,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment