From 37a07a44a1d412444f8c082b6a3a8cee6836e79a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 12 Jun 2019 11:29:36 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index c0880862..f12ba49f 100644 --- a/train.py +++ b/train.py @@ -156,7 +156,7 @@ def train( dataloader = DataLoader(dataset, batch_size=batch_size, num_workers=opt.num_workers, - shuffle=False, # disable rectangular training if True + shuffle=True, # disable rectangular training if True pin_memory=True, collate_fn=dataset.collate_fn)