From 447a3d923d87ca5aba63d6dbdb403b8f154e2652 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 17 Apr 2019 12:42:10 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 5bce4adc..0b3eb4e4 100644 --- a/train.py +++ b/train.py @@ -92,7 +92,7 @@ def train( dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=num_workers, - shuffle=False, + shuffle=True, pin_memory=True, collate_fn=dataset.collate_fn, sampler=sampler)