Anpassungen configs

config.json

@@ -16,7 +16,7 @@
   "out_indices": [0, 1, 2, 3],
   "num_classes": 19,
   "decode_head": {
-    "in_channels": [32, 64, 160, 256],
+    "in_channels": [64, 128, 320, 512],
     "in_index": [0, 1, 2, 3],
     "channels": 256,
     "dropout_ratio": 0.1,

mix_vision_transformer_config.py

@@ -43,7 +43,7 @@ class MySegformerConfig(PretrainedConfig):
         # optionaler Block für Head-Config (falls decode_head nicht übergeben wurde)
         if decode_head is None:
             decode_head = {
-                "in_channels": self.embed_dims,          # [32, 64, 160, 256]
+                "in_channels": [64, 128, 320, 512],          # [32, 64, 160, 256]
                 "in_index": list(range(self.num_stages)),# [0, 1, 2, 3]
                 "channels": 256,
                 "dropout_ratio": 0.1,