is it slurm, or repeat?

mwalmsley · Nov 6, 2023 · 1c0b220 · 1c0b220
1 parent c32524a
commit 1c0b220
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/zoobot/pytorch/training/webdatamodule.py b/zoobot/pytorch/training/webdatamodule.py
@@ -85,7 +85,7 @@ def make_loader(self, urls, mode="train"):
             # torch collate stacks dicts nicely while webdataset only lists them
             # so use the torch collate instead
             .batched(self.batch_size, torch.utils.data.default_collate, partial=False) 
-            # .repeat(5)
+            .repeat(5)
         )
 
         # from itertools import islice