Inria datamodule: needs augs during test stage too

microsoft · Jan 24, 2023 · 671737f · 671737f
1 parent f267766
commit 671737f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"],