File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
segmentation_models_pytorch/decoders/dpt Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -54,10 +54,10 @@ class DPT(SegmentationModel):
54
54
**callable** and **None**. Default is **None**.
55
55
aux_params: Dictionary with parameters of the auxiliary output (classification head). Auxiliary output is build
56
56
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
+ - ** 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).
61
61
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models. Keys with
62
62
``None`` values are pruned before passing. Specify ``dynamic_img_size=True`` to allow the model to handle images of different sizes.
63
63
You can’t perform that action at this time.
0 commit comments