init LOCE

fcjian · Aug 20, 2021 · 09eeed5 · 09eeed5
1 parent 582b874
commit 09eeed5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mmdet/datasets/samplers/distributed_banlanced_sampler.py b/mmdet/datasets/samplers/distributed_banlanced_sampler.py
@@ -27,8 +27,7 @@ def __iter__(self):
             assert len(indices) == self.total_size
 
             # subsample
-            # indices = indices[self.rank:self.total_size:self.num_replicas]
-            indices = indices[self.rank * self.num_samples:self.rank * self.num_samples + self.num_samples]
+            indices = indices[self.rank:self.total_size:self.num_replicas]
             assert len(indices) == self.num_samples
 
             indices_all_classes += indices