add upsampling parameter #1106

Author: DCalhas

segmentation_models_pytorch/decoders/segformer/model.py

@@ -56,6 +56,7 @@ def __init__(
         in_channels: int = 3,
         classes: int = 1,
         activation: Optional[Union[str, Callable]] = None,
+        upsampling: int = 4,
         aux_params: Optional[dict] = None,
         **kwargs: dict[str, Any],
     ):
@@ -80,7 +81,7 @@ def __init__(
             out_channels=classes,
             activation=activation,
             kernel_size=1,
-            upsampling=4,
+            upsampling=upsampling,
         )
 
         if aux_params is not None: