fix(data): fix sampler bug for single device (#22)

Co-authored-by: liusongtao <[email protected]>

exps/example/yolox_voc/yolox_voc_s.py

@@ -58,11 +58,10 @@ class Exp(MyExp):
 
         if is_distributed:
             batch_size = batch_size // dist.get_world_size()
-            sampler = InfiniteSampler(
-                len(self.dataset), seed=self.seed if self.seed else 0
-            )
-        else:
-            sampler = torch.utils.data.RandomSampler(self.dataset)
+
+        sampler = InfiniteSampler(
+            len(self.dataset), seed=self.seed if self.seed else 0
+        )
 
         batch_sampler = YoloBatchSampler(
             sampler=sampler,

yolox/exp/yolox_base.py

@@ -121,11 +121,10 @@ class Exp(BaseExp):
 
         if is_distributed:
             batch_size = batch_size // dist.get_world_size()
-            sampler = InfiniteSampler(
-                len(self.dataset), seed=self.seed if self.seed else 0
-            )
-        else:
-            sampler = torch.utils.data.RandomSampler(self.dataset)
+
+        sampler = InfiniteSampler(
+            len(self.dataset), seed=self.seed if self.seed else 0
+        )
 
         batch_sampler = YoloBatchSampler(
             sampler=sampler,