From c24800b75a137d6420207e0dc46bff816fc88e10 Mon Sep 17 00:00:00 2001 From: Hu Ye Date: Wed, 29 Dec 2021 12:17:26 +0800 Subject: [PATCH] only set random generator if shuffle=true --- references/classification/sampler.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/references/classification/sampler.py b/references/classification/sampler.py index 3c5e8b014b1..e9dc1735a58 100644 --- a/references/classification/sampler.py +++ b/references/classification/sampler.py @@ -36,10 +36,10 @@ def __init__(self, dataset, num_replicas=None, rank=None, shuffle=True, seed=0, self.repetitions = repetitions def __iter__(self): - # Deterministically shuffle based on epoch - g = torch.Generator() - g.manual_seed(self.seed + self.epoch) if self.shuffle: + # Deterministically shuffle based on epoch + g = torch.Generator() + g.manual_seed(self.seed + self.epoch) indices = torch.randperm(len(self.dataset), generator=g).tolist() else: indices = list(range(len(self.dataset)))