fix that transforms are correctly passed to the dataset

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(