update tests

microsoft · Apr 14, 2023 · 7cbef78 · 7cbef78
1 parent f85565f
commit 7cbef78
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/conf/ucmerced.yaml b/tests/conf/ucmerced.yaml
@@ -11,5 +11,5 @@ experiment:
   datamodule:
     root: "tests/data/ucmerced"
     download: true
-    batch_size: 1
+    batch_size: 2
     num_workers: 0
diff --git a/torchgeo/datamodules/ucmerced.py b/torchgeo/datamodules/ucmerced.py
@@ -7,7 +7,7 @@
 
 import kornia.augmentation as K
 import torch
-import torch.nn.functional as F
+import torchvision.transforms.functional as F
 
 from ..datasets import UCMerced
 from ..transforms import AugmentationSequential
@@ -35,7 +35,7 @@ def __init__(
         def default_transform(
             sample: Dict[str, torch.Tensor]
         ) -> Dict[str, torch.Tensor]:
-            sample["image"] = F.interpolate(sample["image"], size=256)
+            sample["image"] = F.resize(sample["image"], size=(256, 256), antialias=True)
             return sample
 
         kwargs["transforms"] = kwargs.get("transforms", default_transform)