Update dataloading.py

lhallee · lhallee · commit d558a17f81bc · 2025-01-06T00:54:10.000-05:00
diff --git a/dataloading.py b/dataloading.py
@@ -75,8 +75,7 @@ def advance(self):
         
         # shuffle each epoch
         if self.next_shard % len(self.files) == 0:
-            self.epoch += 1
-            random.seed(self.epoch)
+            random.seed(self.next_shard)
             random.shuffle(self.files)
 
         processed_chunks = []
