Dont augment val-set

awi-response · Dec 24, 2024 · ef18e6c · ef18e6c
1 parent 9c5b6f1
commit ef18e6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)