.

Dashark · Dashark · Jul 30, 2021 · Jul 30, 2021 · Aug 2, 2021 · Aug 4, 2021
commit 7d4c03d1996d31bf3330d6652ce3604afbe5345b
diff --git a/datasets/SemanticKitti.py b/datasets/SemanticKitti.py
@@ -230,6 +230,8 @@ def __getitem__(self, batch_i):
 
                 # Update epoch indice
                 self.epoch_i += 1
+                if self.epoch_i >= int(self.epoch_inds.shape[0]):
+                    self.epoch_i -= int(self.epoch_inds.shape[0])
 
             s_ind, f_ind = self.all_inds[ind]