From 7fdbc57f9b56b6139865ae89c3a69d62c61449b3 Mon Sep 17 00:00:00 2001 From: HuguesTHOMAS Date: Thu, 29 Jul 2021 18:09:36 +0000 Subject: [PATCH] correcting bug in SemanticKitti --- datasets/SemanticKitti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datasets/SemanticKitti.py b/datasets/SemanticKitti.py index 8403b0d..157adf0 100644 --- a/datasets/SemanticKitti.py +++ b/datasets/SemanticKitti.py @@ -770,7 +770,7 @@ class SemanticKittiSampler(Sampler): else: class_indices = torch.zeros((0,), dtype=torch.int32) while class_indices.shape[0] < class_n: - new_class_inds = torch.randperm(class_potentials.shape[0]) + new_class_inds = torch.randperm(class_potentials.shape[0]).type(torch.int32) class_indices = torch.cat((class_indices, new_class_inds), dim=0) class_indices = class_indices[:class_n] class_indices = self.dataset.class_frames[i][class_indices]