diff --git a/datasets/SemanticKitti.py b/datasets/SemanticKitti.py
index d2860b5..8eae642 100644
--- a/datasets/SemanticKitti.py
+++ b/datasets/SemanticKitti.py
@@ -230,6 +230,8 @@ class SemanticKittiDataset(PointCloudDataset):
 
                 # 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]