From 595694cb62efe6951da3a3c037b9c94f8de81feb Mon Sep 17 00:00:00 2001
From: Himanshu Pandey <hpandey@pivotal.io>
Date: Fri, 15 Feb 2019 01:26:46 -0800
Subject: [PATCH]  Replaced hard coding for shuffle variable (#438)

---
 maskrcnn_benchmark/data/samplers/distributed.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/maskrcnn_benchmark/data/samplers/distributed.py b/maskrcnn_benchmark/data/samplers/distributed.py
index 5de06e3..27a280f 100644
--- a/maskrcnn_benchmark/data/samplers/distributed.py
+++ b/maskrcnn_benchmark/data/samplers/distributed.py
@@ -37,7 +37,7 @@ class DistributedSampler(Sampler):
         self.epoch = 0
         self.num_samples = int(math.ceil(len(self.dataset) * 1.0 / self.num_replicas))
         self.total_size = self.num_samples * self.num_replicas
-        self.shuffle = True
+        self.shuffle = shuffle
 
     def __iter__(self):
         if self.shuffle:
-- 
GitLab