diff --git a/utils/dataloader.py b/utils/dataloader.py
index 46f9ebc6b8ea851c1919c9f5dae8b4abd2659773..409c6ec24e133ff71b03638de2646e78b2650bae 100644
--- a/utils/dataloader.py
+++ b/utils/dataloader.py
@@ -207,6 +207,7 @@ class YoloDataset(Dataset):
             shuffle(self.train_lines)
         lines = self.train_lines
         n = self.train_batches
+        index = index % n
         if self.mosaic:
             if self.flag and (index + 4) < n:
                 img, y = self.get_random_data_with_Mosaic(lines[index:index + 4], self.image_size[0:2])