Skip to content

Commit 6207310

Browse files
committed
Fixing list
1 parent 0ed621c commit 6207310

File tree

1 file changed

+5
-4
lines changed
  • segmentation_models_pytorch/decoders/dpt

1 file changed

+5
-4
lines changed

segmentation_models_pytorch/decoders/dpt/model.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,11 @@ class DPT(SegmentationModel):
5454
**callable** and **None**. Default is **None**.
5555
aux_params: Dictionary with parameters of the auxiliary output (classification head). Auxiliary output is build
5656
on top of encoder if **aux_params** is not **None** (default). Supported params:
57-
- **classes** (*int*): A number of classes;
58-
- **pooling** (*str*): One of "max", "avg". Default is "avg";
59-
- **dropout** (*float*): Dropout factor in [0, 1);
60-
- **activation** (*str*): An activation function to apply "sigmoid"/"softmax" (could be **None** to return logits).
57+
58+
- **classes** (*int*): A number of classes;
59+
- **pooling** (*str*): One of "max", "avg". Default is "avg";
60+
- **dropout** (*float*): Dropout factor in [0, 1);
61+
- **activation** (*str*): An activation function to apply "sigmoid"/"softmax" (could be **None** to return logits).
6162
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models. Keys with
6263
``None`` values are pruned before passing. Specify ``dynamic_img_size=True`` to allow the model to handle images of different sizes.
6364

0 commit comments

Comments
 (0)