train.py: shuffling at every epoch

NVIDIA · Mar 16, 2019 · f37998c · f37998c
1 parent bff304f
commit f37998c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -59,7 +59,7 @@ def prepare_dataloaders(hparams):
     train_sampler = DistributedSampler(trainset) \
         if hparams.distributed_run else None
 
-    train_loader = DataLoader(trainset, num_workers=1, shuffle=False,
+    train_loader = DataLoader(trainset, num_workers=1, shuffle=True,
                               sampler=train_sampler,
                               batch_size=hparams.batch_size, pin_memory=False,
                               drop_last=True, collate_fn=collate_fn)