From 338520daf89080be3080ad8df2f9a47d20b7bc8c Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Tue, 4 Oct 2022 11:46:24 +0200
Subject: [PATCH] fix that transforms are correctly passed to the dataset
---
mu_map/training/default.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/mu_map/training/default.py b/mu_map/training/default.py
index 8a8e4e4..754a878 100644
--- a/mu_map/training/default.py
+++ b/mu_map/training/default.py
@@ -265,7 +265,8 @@ if __name__ == "__main__":
dataset = MuMapPatchDataset(
args.dataset_dir,
split_name=split,
- transform_normalization=MeanNormTransform(),
+ transform_normalization=transform_normalization,
+ transform_augmentation=transform_augmentation,
logger=logger,
)
data_loader = torch.utils.data.DataLoader(
--
GitLab