from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_tf_available,
    is_torch_available,
    is_vision_available,
)


__lowercase							 =					{'''configuration_deit''': ['''DEIT_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''DeiTConfig''', '''DeiTOnnxConfig''']}

try:
    if not is_vision_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    pass
else:
    __lowercase							 =					['''DeiTFeatureExtractor''']
    __lowercase							 =					['''DeiTImageProcessor''']

try:
    if not is_torch_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    pass
else:
    __lowercase							 =					[
        '''DEIT_PRETRAINED_MODEL_ARCHIVE_LIST''',
        '''DeiTForImageClassification''',
        '''DeiTForImageClassificationWithTeacher''',
        '''DeiTForMaskedImageModeling''',
        '''DeiTModel''',
        '''DeiTPreTrainedModel''',
    ]

try:
    if not is_tf_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    pass
else:
    __lowercase							 =					[
        '''TF_DEIT_PRETRAINED_MODEL_ARCHIVE_LIST''',
        '''TFDeiTForImageClassification''',
        '''TFDeiTForImageClassificationWithTeacher''',
        '''TFDeiTForMaskedImageModeling''',
        '''TFDeiTModel''',
        '''TFDeiTPreTrainedModel''',
    ]


if TYPE_CHECKING:
    from .configuration_deit import DEIT_PRETRAINED_CONFIG_ARCHIVE_MAP, DeiTConfig, DeiTOnnxConfig

    try:
        if not is_vision_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        pass
    else:
        from .feature_extraction_deit import DeiTFeatureExtractor
        from .image_processing_deit import DeiTImageProcessor

    try:
        if not is_torch_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        pass
    else:
        from .modeling_deit import (
            DEIT_PRETRAINED_MODEL_ARCHIVE_LIST,
            DeiTForImageClassification,
            DeiTForImageClassificationWithTeacher,
            DeiTForMaskedImageModeling,
            DeiTModel,
            DeiTPreTrainedModel,
        )

    try:
        if not is_tf_available():
            raise OptionalDependencyNotAvailable()
    except OptionalDependencyNotAvailable:
        pass
    else:
        from .modeling_tf_deit import (
            TF_DEIT_PRETRAINED_MODEL_ARCHIVE_LIST,
            TFDeiTForImageClassification,
            TFDeiTForImageClassificationWithTeacher,
            TFDeiTForMaskedImageModeling,
            TFDeiTModel,
            TFDeiTPreTrainedModel,
        )


else:
    import sys

    __lowercase							 =					_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)







import copy

from ...configuration_utils import PretrainedConfig
from ...utils import logging
from ..auto.configuration_auto import CONFIG_MAPPING


__a  			=					logging.get_logger(__name__)
class  lowercase__(				UpperCAmelCase							):



	"""simple docstring"""




	a			:Union[str, Any]         =	'upernet'
	def __init__(       self  :			Dict       ,							SCREAMING_SNAKE_CASE_  :			Dict=None       ,							SCREAMING_SNAKE_CASE_  :			str=5_1_2       ,							SCREAMING_SNAKE_CASE_  :			Tuple=0.02       ,							SCREAMING_SNAKE_CASE_  :			Optional[Any]=[1, 2, 3, 6]       ,							SCREAMING_SNAKE_CASE_  :			Optional[int]=True       ,							SCREAMING_SNAKE_CASE_  :			Tuple=0.4       ,							SCREAMING_SNAKE_CASE_  :			Optional[int]=3_8_4       ,							SCREAMING_SNAKE_CASE_  :			str=2_5_6       ,							SCREAMING_SNAKE_CASE_  :			Dict=1       ,							SCREAMING_SNAKE_CASE_  :			Tuple=False       ,							SCREAMING_SNAKE_CASE_  :			str=2_5_5       ,							**SCREAMING_SNAKE_CASE_  :			str       ,							)       ->     int:
					super().__init__(**SCREAMING_SNAKE_CASE_       )

					if backbone_config is None:
									logger.info('''`backbone_config` is `None`. Initializing the config with the default `ResNet` backbone.'''       )
									lowercase_       =					CONFIG_MAPPING['''resnet'''](out_features=['''stage1''', '''stage2''', '''stage3''', '''stage4''']       )
					elif isinstance(SCREAMING_SNAKE_CASE_       ,							SCREAMING_SNAKE_CASE_       ):
									lowercase_       =					backbone_config.get('''model_type'''       )
									lowercase_       =					CONFIG_MAPPING[backbone_model_type]
									lowercase_       =					config_class.from_dict(SCREAMING_SNAKE_CASE_       )

					lowercase_       =					backbone_config
					lowercase_       =					hidden_size
					lowercase_       =					initializer_range
					lowercase_       =					pool_scales
					lowercase_       =					use_auxiliary_head
					lowercase_       =					auxiliary_loss_weight
					lowercase_       =					auxiliary_in_channels
					lowercase_       =					auxiliary_channels
					lowercase_       =					auxiliary_num_convs
					lowercase_       =					auxiliary_concat_input
					lowercase_       =					loss_ignore_index




	def 						_lowercase  (       self  :			List[str]       )       ->     List[str]:
					lowercase_       =					copy.deepcopy(self.__dict__       )
					lowercase_       =					self.backbone_config.to_dict()
					lowercase_       =					self.__class__.model_type
					return output




