diff --git a/mmdet/datasets/loader/sampler.py b/mmdet/datasets/loader/sampler.py index 6c529008249149ae8d90a2f0b427a74f97b98048..c6109388d13bfad0c20b102b8d4735ba401de848 100644 --- a/mmdet/datasets/loader/sampler.py +++ b/mmdet/datasets/loader/sampler.py @@ -132,8 +132,12 @@ class DistributedGroupSampler(Sampler): math.ceil( size * 1.0 / self.samples_per_gpu / self.num_replicas) ) * self.samples_per_gpu * self.num_replicas - len(indice) - indice += indice[:extra] - indices += indice + # pad indice + tmp = indice.copy() + for _ in range(extra // size): + indice.extend(tmp) + indice.extend(tmp[:extra % size]) + indices.extend(indice) assert len(indices) == self.total_size