diff --git a/torchgeo/datamodules/inria.py b/torchgeo/datamodules/inria.py index bc524e5e921..81503810011 100644 --- a/torchgeo/datamodules/inria.py +++ b/torchgeo/datamodules/inria.py @@ -58,7 +58,7 @@ def __init__( _RandomNCrop(self.patch_size, batch_size), data_keys=["image", "mask"], ) - self.val_aug = AugmentationSequential( + self.aug = AugmentationSequential( K.Normalize(mean=self.mean, std=self.std), _RandomNCrop(self.patch_size, batch_size), data_keys=["image", "mask"],