Anpassung in_channels

modeling_my_segformer.py

@@ -32,7 +32,7 @@ class MySegformerForSemanticSegmentation(PreTrainedModel):
         )
 
         # Head direkt importieren
-        in_channels = [32, 64, 160, 256]
+        in_channels = [64, 128, 320, 512]
 
         self.segmentation_head = SegformerHead(
             in_channels=in_channels,                # Liste der Embeddings aus Backbone