From 1c0b220335da60a9f9d1d698c4b456842c7f4387 Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Mon, 6 Nov 2023 14:04:24 -0500 Subject: [PATCH] is it slurm, or repeat? --- zoobot/pytorch/training/webdatamodule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/zoobot/pytorch/training/webdatamodule.py b/zoobot/pytorch/training/webdatamodule.py index be51cbc3..a965a777 100644 --- 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