Skip to content
Snippets Groups Projects
Commit 084a3890 authored by Jiaqi Wang's avatar Jiaqi Wang Committed by Kai Chen
Browse files

fix a potential bug in GroupSampler (#955)

parent b8bcda67
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment