Tim77777767
commited on
Commit
·
9659a3a
1
Parent(s):
e027211
Debug eingefügt
Browse files- modeling_my_segformer.py +4 -0
modeling_my_segformer.py
CHANGED
|
@@ -50,6 +50,10 @@ class MySegformerForSemanticSegmentation(PreTrainedModel):
|
|
| 50 |
# Backbone → Features (Liste von Tensors)
|
| 51 |
features = self.backbone(x)
|
| 52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
# Head → logits
|
| 54 |
logits = self.segmentation_head(features)
|
| 55 |
|
|
|
|
| 50 |
# Backbone → Features (Liste von Tensors)
|
| 51 |
features = self.backbone(x)
|
| 52 |
|
| 53 |
+
# Debug: Ausgabe der Shapes der Backbone-Features
|
| 54 |
+
for i, f in enumerate(features):
|
| 55 |
+
print(f"Feature {i}: shape = {f.shape}")
|
| 56 |
+
|
| 57 |
# Head → logits
|
| 58 |
logits = self.segmentation_head(features)
|
| 59 |
|