dim typo

microsoft · Sep 24, 2023 · 13f8324 · 13f8324
1 parent edc4433
commit 13f8324
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/datasets/skippd.py b/torchgeo/datasets/skippd.py
@@ -166,7 +166,7 @@ def _load_image(self, index: int) -> Tensor:
         ) as f:
             arr = f[self.split]["images_log"][index]
 
-        # forecast has dimension [16, 64, 64, 3] but reshape to [64, 64, 48]
+        # forecast has dimension [16, 64, 64, 3] but reshape to [48, 64, 64]
         # https://github.com/yuhao-nie/Stanford-solar-forecasting-dataset/blob/main/models/SUNSET_forecast.ipynb
         if self.task == "forecast":
             arr = rearrange(arr, "t h w c-> (t c) h w")