Fix issue #7209 in DataLoader

vfdev-5 · vfdev-5 · commit 7ecf31529beb · 2018-05-04T00:53:34.000+02:00
diff --git a/torch/utils/data/dataloader.py b/torch/utils/data/dataloader.py
@@ -246,6 +246,8 @@ def __init__(self, loader):
 
         self.sample_iter = iter(self.batch_sampler)
 
+        base_seed = torch.LongTensor(1).random_()[0]
+
         if self.num_workers > 0:
             self.worker_init_fn = loader.worker_init_fn
             self.index_queues = [multiprocessing.Queue() for _ in range(self.num_workers)]
@@ -258,7 +260,6 @@ def __init__(self, loader):
             self.rcvd_idx = 0
             self.reorder_dict = {}
 
-            base_seed = torch.LongTensor(1).random_()[0]
             self.workers = [
                 multiprocessing.Process(
                     target=_worker_loop,
