Commit 15237cc4 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

merge and sidesampler

parent 28205f33
......@@ -1087,7 +1087,7 @@ def get_loaders(dataset_opts, training_opts, model_opts, local_rank=0):
)
training_loader = DataLoader(training_set,
batch_size=batch_size,
batch_size=batch_size * dataset_opts["train"]["sampler"]["augmentation_replicas"],
shuffle=False,
drop_last=True,
pin_memory=True,
......
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