restored aux_params

segmentation_models_pytorch/decoders/deeplabv3/model.py

@@ -145,6 +145,7 @@ class DeepLabV3Plus(SegmentationModel):
                 **callable** and **None**.
             Default is **None**
         upsampling: Final upsampling factor. Default is 4 to preserve input-output spatial shape identity.
+        aux_params: Dictionary with parameters of the auxiliary output (classification head). Auxiliary output is build
             on top of encoder if **aux_params** is not **None** (default). Supported params:
                 - classes (int): A number of classes
                 - pooling (str): One of "max", "avg". Default is "avg"