Anpassungen Head

segformer_plusplus/model/head/segformer_head.py

@@ -49,6 +49,10 @@ class SegformerHead(BaseModule):
 
         num_inputs = len(self.in_channels)
 
+        print(num_inputs)
+        print(in_index)
+        #assert num_inputs == len(self.in_index)
+
         self.convs = nn.ModuleList()
         for i in range(num_inputs):
             self.convs.append(