Anpassungen Configs

config.json

@@ -1,7 +1,7 @@
 {
   "model_type": "my_segformer",
   "in_channels": 3,
-  "embed_dims": [32, 64, 160, 256],
+  "embed_dims": 64,
   "num_stages": 4,
   "num_layers": [2, 2, 2, 2],
   "num_heads": [1, 2, 5, 8],

modeling_my_segformer.py

@@ -32,9 +32,7 @@ class MySegformerForSemanticSegmentation(PreTrainedModel):
         )
 
         # Head direkt importieren
-        in_channels = config.embed_dims
-        if isinstance(in_channels, int):
-            in_channels = [in_channels]
+        in_channels = [32, 64, 160, 256]
 
         self.segmentation_head = SegformerHead(
             in_channels=in_channels,                # Liste der Embeddings aus Backbone