fix a potential bug in GroupSampler (#955)

084a3890 · Jiaqi Wang · Kai Chen · b8bcda67 · 084a3890
Commit 084a3890 authored Jul 08, 2019 by Jiaqi Wang Committed by Kai Chen Jul 08, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet/datasets/loader/sampler.py mmdet/datasets/loader/sampler.py +1 -1

No files found.
--- a/mmdet/datasets/loader/sampler.py
+++ b/mmdet/datasets/loader/sampler.py
@@ -67,7 +67,7 @@ class GroupSampler(Sampler):
                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)