From 8d9c8e0622e80a3c56fd232b681185e04345d02e Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 18 Apr 2018 13:26:17 +0200
Subject: k

---
 data/__init__.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'data')

diff --git a/data/__init__.py b/data/__init__.py
index 381ec59..6da79f3 100644
--- a/data/__init__.py
+++ b/data/__init__.py
@@ -70,7 +70,11 @@ class RecursiveDatasetDataLoader(BaseDataLoader):
         #BaseDataLoader.initialize(self, opt)
         self.opt = opt
         self.dataset = CreateDataset(opt)
-        self.dataloader = self.dataset
+        self.dataloader = torch.utils.data.DataLoader(
+            self.dataset,
+            batch_size=opt.batchSize,
+            shuffle=not opt.serial_batches,
+            num_workers=int(opt.nThreads))
 
     def load_data(self):
         return self
-- 
cgit v1.2.3-70-g09d2