Tim77777767
commited on
Commit
·
a489aa9
1
Parent(s):
18749b6
Anpüassung am Head
Browse files
segformer_plusplus/model/head/segformer_head.py
CHANGED
@@ -49,8 +49,6 @@ class SegformerHead(BaseModule):
|
|
49 |
|
50 |
num_inputs = len(self.in_channels)
|
51 |
|
52 |
-
assert num_inputs == len(self.in_index)
|
53 |
-
|
54 |
self.convs = nn.ModuleList()
|
55 |
for i in range(num_inputs):
|
56 |
self.convs.append(
|
|
|
49 |
|
50 |
num_inputs = len(self.in_channels)
|
51 |
|
|
|
|
|
52 |
self.convs = nn.ModuleList()
|
53 |
for i in range(num_inputs):
|
54 |
self.convs.append(
|