From ef18e6cc4653300329174d159da8a0d48d4917f4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tobias=20H=C3=B6lzer?= Date: Tue, 24 Dec 2024 21:10:03 +0100 Subject: [PATCH] Dont augment val-set --- darts-segmentation/src/darts_segmentation/training/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/darts-segmentation/src/darts_segmentation/training/data.py b/darts-segmentation/src/darts_segmentation/training/data.py index 5e1698c..b416ad1 100644 --- a/darts-segmentation/src/darts_segmentation/training/data.py +++ b/darts-segmentation/src/darts_segmentation/training/data.py @@ -146,7 +146,7 @@ def setup(self, stage: Literal["fit", "validate", "test", "predict"] | None = No train_idx, val_idx = list(kf.split(dataset))[self.current_fold] self.train = Subset(dataset, train_idx) self.val = Subset(dataset, val_idx) - self.val.dataset.transform = False + self.val.dataset.transform = None def train_dataloader(self): return DataLoader(self.train, batch_size=self.batch_size, num_workers=self.num_workers, shuffle=True)