Skip to content

Commit bc9cde2

Browse files
committed
add upsampling parameter qubvel-org#1106
1 parent 9532612 commit bc9cde2

File tree

1 file changed

+2
-1
lines changed
  • segmentation_models_pytorch/decoders/segformer

1 file changed

+2
-1
lines changed

segmentation_models_pytorch/decoders/segformer/model.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ def __init__(
5656
in_channels: int = 3,
5757
classes: int = 1,
5858
activation: Optional[Union[str, Callable]] = None,
59+
upsampling: int = 4,
5960
aux_params: Optional[dict] = None,
6061
**kwargs: dict[str, Any],
6162
):
@@ -80,7 +81,7 @@ def __init__(
8081
out_channels=classes,
8182
activation=activation,
8283
kernel_size=1,
83-
upsampling=4,
84+
upsampling=upsampling,
8485
)
8586

8687
if aux_params is not None:

0 commit comments

Comments
 (0)