Merge pull request #120 from yurujaja/119-fix-gfmswin

119 fix gfmswin

configs/decoder/seg_upernet_mt_none.yaml

@@ -0,0 +1,11 @@
+_target_: pangaea.decoders.upernet.SegMTUPerNet
+
+# overwritten in run
+encoder: null
+num_classes: ${dataset.num_classes}
+multi_temporal: ${dataset.multi_temporal}
+finetune: ${finetune}
+
+# to adapt
+channels: 512
+multi_temporal_strategy: none

pangaea/decoders/upernet.py

@@ -277,6 +277,7 @@ def __init__(
         # returns time merged outputs then we don't need multi_temporal_strategy
         if self.encoder.multi_temporal and not self.encoder.multi_temporal_output:
             self.tmap = None
+
         else:
             if self.multi_temporal_strategy == "ltae":
                 ltae_in_channels = max(decoder_in_channels)
@@ -302,7 +303,7 @@ def __init__(
     def get_decoder_in_channels(
         self, multi_temporal_strategy: str | None, encoder: Encoder
     ) -> list[int]:
-        if multi_temporal_strategy == "ltae" and encoder.multi_temporal_output:
+        if multi_temporal_strategy == "ltae":
             # if the encoder output channels vary we must use an adaptor before the LTAE
             ltae_in_channels = max(encoder.output_dim)
             if ltae_in_channels != min(encoder.output_dim):