Anpassung an den 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],

mix_vision_transformer_config.py

@@ -24,11 +24,7 @@ class MySegformerConfig(PretrainedConfig):
         **kwargs
     ):
         super().__init__(**kwargs)
-
-        # Falls embed_dims als int angegeben ist, Liste erzwingen
-        if isinstance(embed_dims, int):
-            embed_dims = [embed_dims]
-
+        self.embed_dims = embed_dims  # einfach int behalten
         self.in_channels = in_channels
         self.embed_dims = embed_dims
         self.num_stages = num_stages