From 5fe0346176235520ba42a1271235e30ca1f9d7ba Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 17 Apr 2019 18:40:12 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index bbfb55f5..6bd361bb 100644 --- a/train.py +++ b/train.py @@ -113,7 +113,7 @@ def train( # Dataloader dataloader = DataLoader(dataset, batch_size=batch_size, - num_workers=0, + num_workers=opt.num_workers, shuffle=True, pin_memory=True, collate_fn=dataset.collate_fn,