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