Skip to content

Commit 463ee59

Browse files
committed
docstring upsampling parameter #1123
1 parent bc9cde2 commit 463ee59

File tree

1 file changed

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

1 file changed

+1
-0
lines changed

segmentation_models_pytorch/decoders/segformer/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ class Segformer(SegmentationModel):
2929
activation: An activation function to apply after the final convolution layer.
3030
Available options are **"sigmoid"**, **"softmax"**, **"logsoftmax"**, **"tanh"**, **"identity"**,
3131
**callable** and **None**. Default is **None**.
32+
upsampling: A number to upsample the output of the model, default is 4 (same size as input)
3233
aux_params: Dictionary with parameters of the auxiliary output (classification head). Auxiliary output is build
3334
on top of encoder if **aux_params** is not **None** (default). Supported params:
3435
- classes (int): A number of classes

0 commit comments

Comments
 (0)