U
    ×9%ed  ã                   @   sF   d Z ddlmZ ddlmZ ddlmZ e e¡Z	G dd„ deƒZ
dS )	z UperNet model configurationé   )ÚPretrainedConfig)Úloggingé   )ÚCONFIG_MAPPINGc                       sD   e Zd ZdZdZddddddd	gd
ddddddf‡ fdd„	Z‡  ZS )ÚUperNetConfigaÖ	  
    This is the configuration class to store the configuration of an [`UperNetForSemanticSegmentation`]. It is used to
    instantiate an UperNet model according to the specified arguments, defining the model architecture. Instantiating a
    configuration with the defaults will yield a similar configuration to that of the UperNet
    [openmmlab/upernet-convnext-tiny](https://huggingface.co/openmmlab/upernet-convnext-tiny) architecture.

    Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
    documentation from [`PretrainedConfig`] for more information.

    Args:
        backbone_config (`PretrainedConfig` or `dict`, *optional*, defaults to `ResNetConfig()`):
            The configuration of the backbone model.
        hidden_size (`int`, *optional*, defaults to 512):
            The number of hidden units in the convolutional layers.
        initializer_range (`float`, *optional*, defaults to 0.02):
            The standard deviation of the truncated_normal_initializer for initializing all weight matrices.
        pool_scales (`Tuple[int]`, *optional*, defaults to `[1, 2, 3, 6]`):
            Pooling scales used in Pooling Pyramid Module applied on the last feature map.
        use_auxiliary_head (`bool`, *optional*, defaults to `True`):
            Whether to use an auxiliary head during training.
        auxiliary_loss_weight (`float`, *optional*, defaults to 0.4):
            Weight of the cross-entropy loss of the auxiliary head.
        auxiliary_channels (`int`, *optional*, defaults to 256):
            Number of channels to use in the auxiliary head.
        auxiliary_num_convs (`int`, *optional*, defaults to 1):
            Number of convolutional layers to use in the auxiliary head.
        auxiliary_concat_input (`bool`, *optional*, defaults to `False`):
            Whether to concatenate the output of the auxiliary head with the input before the classification layer.
        loss_ignore_index (`int`, *optional*, defaults to 255):
            The index that is ignored by the loss function.

    Examples:

    ```python
    >>> from transformers import UperNetConfig, UperNetForSemanticSegmentation

    >>> # Initializing a configuration
    >>> configuration = UperNetConfig()

    >>> # Initializing a model (with random weights) from the configuration
    >>> model = UperNetForSemanticSegmentation(configuration)

    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```ZupernetNi   g{®Gáz”?é   r   r   é   Tgš™™™™™Ù?i€  é   Féÿ   c                    s¤   t ƒ jf |Ž |d kr8t d¡ td ddddgd}n&t|tƒr^| d¡}t| }| |¡}|| _	|| _
|| _|| _|| _|| _|| _|| _|	| _|
| _|| _d S )	NzX`backbone_config` is `None`. Initializing the config with the default `ResNet` backbone.ZresnetZstage1Zstage2Zstage3Zstage4)Zout_featuresÚ
model_type)ÚsuperÚ__init__ÚloggerÚinfor   Ú
isinstanceÚdictÚgetÚ	from_dictÚbackbone_configÚhidden_sizeÚinitializer_rangeÚpool_scalesÚuse_auxiliary_headÚauxiliary_loss_weightÚauxiliary_in_channelsÚauxiliary_channelsÚauxiliary_num_convsÚauxiliary_concat_inputÚloss_ignore_index)Úselfr   r   r   r   r   r   r   r   r   r   r   ÚkwargsZbackbone_model_typeZconfig_class©Ú	__class__© úp/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/transformers/models/upernet/configuration_upernet.pyr   J   s&    



zUperNetConfig.__init__)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   Ú__classcell__r#   r#   r!   r$   r      s   -
ôr   N)r(   Zconfiguration_utilsr   Úutilsr   Zauto.configuration_autor   Z
get_loggerr%   r   r   r#   r#   r#   r$   Ú<module>   s
   
