Fix

qubvel-org · Jan 14, 2025 · 4f65d8f · 4f65d8f
1 parent 3cb8198
commit 4f65d8f
Showing 1 changed file with 1 addition and 1 deletion.
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]