Skip to content

Commit f96dbda

Browse files
committed
update deeplabv3+ doc
1 parent bafd4fb commit f96dbda

File tree

1 file changed

+1
-3
lines changed
  • segmentation_models_pytorch/decoders/deeplabv3

1 file changed

+1
-3
lines changed

segmentation_models_pytorch/decoders/deeplabv3/model.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -145,9 +145,7 @@ class DeepLabV3Plus(SegmentationModel):
145145
Available options are **"sigmoid"**, **"softmax"**, **"logsoftmax"**, **"tanh"**, **"identity"**,
146146
**callable** and **None**.
147147
Default is **None**
148-
upsampling: Final upsampling factor. Default is 4 to preserve input-output spatial shape identity. In case
149-
**encoder_depth** and **encoder_output_stride** are 3 and 16 resp., set **upsampling** to 2 to preserve.
150-
aux_params: Dictionary with parameters of the auxiliary output (classification head). Auxiliary output is build
148+
upsampling: Final upsampling factor. Default is 4 to preserve input-output spatial shape identity.
151149
on top of encoder if **aux_params** is not **None** (default). Supported params:
152150
- classes (int): A number of classes
153151
- pooling (str): One of "max", "avg". Default is "avg"

0 commit comments

Comments
 (0)