From 7ecf31529beb02e7afda25f5760e94e4451c05f1 Mon Sep 17 00:00:00 2001 From: vfdev-5 Date: Thu, 3 May 2018 22:47:20 +0200 Subject: [PATCH] Fix issue #7209 in DataLoader --- torch/utils/data/dataloader.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torch/utils/data/dataloader.py b/torch/utils/data/dataloader.py index 027f23a1a6bdf..5678c3a0f2de3 100644 --- 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,