Commit f415dc85 authored by mpatwary's avatar mpatwary
Browse files

removed commnets

parent 80047314
......@@ -90,8 +90,6 @@ def build_data_loader(dataset, micro_batch_size, num_workers, drop_last,
sampler = torch.utils.data.distributed.DistributedSampler(
dataset, num_replicas=world_size, rank=rank)
print_rank_0(len(sampler))
# Data loader. Note that batch size is the per GPU batch size.
data_loader = torch.utils.data.DataLoader(dataset,
batch_size=micro_batch_size,
......
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