fix a potential bug in GroupSampler (#955)

open-mmlab · Jul 8, 2019 · 084a389 · 084a389
1 parent b8bcda6
commit 084a389
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/datasets/loader/sampler.py b/mmdet/datasets/loader/sampler.py
@@ -67,7 +67,7 @@ def __iter__(self):
                 range(len(indices) // self.samples_per_gpu))
         ]
         indices = np.concatenate(indices)
-        indices = torch.from_numpy(indices).long()
+        indices = indices.astype(np.int64).tolist()
         assert len(indices) == self.num_samples
         return iter(indices)