From 13f83249fb1f60007293b95558be1f38744c4fed Mon Sep 17 00:00:00 2001 From: nilsleh Date: Sun, 24 Sep 2023 16:35:52 +0000 Subject: [PATCH] dim typo --- torchgeo/datasets/skippd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchgeo/datasets/skippd.py b/torchgeo/datasets/skippd.py index aa64d3081b2..1242678c8b9 100644 --- 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")