From 3cde2de73e262114ac906e3f876f3d6a07674037 Mon Sep 17 00:00:00 2001 From: Ashwin Nair Date: Thu, 4 Jul 2024 11:08:24 +0400 Subject: [PATCH] Add issue link --- torchgeo/datamodules/nasa_marine_debris.py | 1 + torchgeo/datamodules/vhr10.py | 1 + 2 files changed, 2 insertions(+) diff --git a/torchgeo/datamodules/nasa_marine_debris.py b/torchgeo/datamodules/nasa_marine_debris.py index 2fcba2236c3..add6b32b063 100644 --- a/torchgeo/datamodules/nasa_marine_debris.py +++ b/torchgeo/datamodules/nasa_marine_debris.py @@ -48,6 +48,7 @@ def __init__( self.aug = K.AugmentationSequential( K.Normalize(mean=self.mean, std=self.std), data_keys=None, keepdim=True ) + # https://github.com/kornia/kornia/issues/2848 self.aug.keepdim = True # type: ignore[attr-defined] self.collate_fn = collate_fn_detection diff --git a/torchgeo/datamodules/vhr10.py b/torchgeo/datamodules/vhr10.py index 49e7b90241b..8e9d56c7f7a 100644 --- a/torchgeo/datamodules/vhr10.py +++ b/torchgeo/datamodules/vhr10.py @@ -64,6 +64,7 @@ def __init__( self.aug = K.AugmentationSequential( K.Normalize(mean=self.mean, std=self.std), data_keys=None, keepdim=True ) + # https://github.com/kornia/kornia/issues/2848 self.aug.keepdim = True # type: ignore[attr-defined] def setup(self, stage: str) -> None: