Add issue link

microsoft · Jul 4, 2024 · 3cde2de · 3cde2de
1 parent 4b199cd
commit 3cde2de
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -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: