Fix

qubvel · qubvel · commit 4f65d8f1e4a8 · 2025-01-14T13:08:01.000Z
diff --git a/segmentation_models_pytorch/decoders/upernet/decoder.py b/segmentation_models_pytorch/decoders/upernet/decoder.py
@@ -110,7 +110,7 @@ def __init__(
             use_batchnorm=True,
         )
 
-    def forward(self, *features):
+    def forward(self, features):
         output_size = features[0].shape[2:]
         target_size = [size // 4 for size in output_size]
 

Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,7 @@ def __init__(`
`110`	`110`	`use_batchnorm=True,`
`111`	`111`	`)`
`112`	`112`
`113`		`- def forward(self, *features):`
	`113`	`+ def forward(self, features):`
`114`	`114`	`output_size = features[0].shape[2:]`
`115`	`115`	`target_size = [size // 4 for size in output_size]`
`116`	`116`