From 671737fdbc3c24557019fde4973ed3bd2794d380 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Tue, 24 Jan 2023 16:08:52 -0600 Subject: [PATCH] Inria datamodule: needs augs during test stage too --- torchgeo/datamodules/inria.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"],