From 9fc8a4b35a0483e9215e1d8eb026f461d86fba28 Mon Sep 17 00:00:00 2001
From: Ryan <23580140+brianhou0208@users.noreply.github.com>
Date: Sat, 30 Nov 2024 23:26:36 +0800
Subject: [PATCH] restored aux_params

---
 segmentation_models_pytorch/decoders/deeplabv3/model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/segmentation_models_pytorch/decoders/deeplabv3/model.py b/segmentation_models_pytorch/decoders/deeplabv3/model.py
index 53e1846e..830906cb 100644
--- a/segmentation_models_pytorch/decoders/deeplabv3/model.py
+++ b/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"