U ,-e@sBdZddlmZddlmZeeZddiZGdddeZ dS) z RegNet model configuration)PretrainedConfig)loggingzfacebook/regnet-y-040zBhttps://huggingface.co/facebook/regnet-y-040/blob/main/config.jsoncsLeZdZdZdZddgZddddd d gd d d d gdddffdd ZZS) RegNetConfiga This is the configuration class to store the configuration of a [`RegNetModel`]. It is used to instantiate a RegNet 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 RegNet [facebook/regnet-y-040](https://huggingface.co/facebook/regnet-y-040) architecture. Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the documentation from [`PretrainedConfig`] for more information. Args: num_channels (`int`, *optional*, defaults to 3): The number of input channels. embedding_size (`int`, *optional*, defaults to 64): Dimensionality (hidden size) for the embedding layer. hidden_sizes (`List[int]`, *optional*, defaults to `[256, 512, 1024, 2048]`): Dimensionality (hidden size) at each stage. depths (`List[int]`, *optional*, defaults to `[3, 4, 6, 3]`): Depth (number of layers) for each stage. layer_type (`str`, *optional*, defaults to `"y"`): The layer to use, it can be either `"x" or `"y"`. An `x` layer is a ResNet's BottleNeck layer with `reduction` fixed to `1`. While a `y` layer is a `x` but with squeeze and excitation. Please refer to the paper for a detailed explanation of how these layers were constructed. hidden_act (`str`, *optional*, defaults to `"relu"`): The non-linear activation function in each block. If string, `"gelu"`, `"relu"`, `"selu"` and `"gelu_new"` are supported. downsample_in_first_stage (`bool`, *optional*, defaults to `False`): If `True`, the first stage will downsample the inputs using a `stride` of 2. Example: ```python >>> from transformers import RegNetConfig, RegNetModel >>> # Initializing a RegNet regnet-y-40 style configuration >>> configuration = RegNetConfig() >>> # Initializing a model from the regnet-y-40 style configuration >>> model = RegNetModel(configuration) >>> # Accessing the model configuration >>> configuration = model.config ``` Zregnetxyr ii@ @Zreluc  shtjf|||jkr4td|dd|j||_||_||_||_||_ ||_ ||_ d|_ dS)Nz layer_type=z is not one of ,T) super__init__ layer_types ValueErrorjoin num_channelsembedding_size hidden_sizesdepths groups_width layer_type hidden_actZdownsample_in_first_stage) selfrrrrrrrkwargs __class__p/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/transformers/models/regnet/configuration_regnet.pyrHs  zRegNetConfig.__init__)__name__ __module__ __qualname____doc__Z model_typerr __classcell__rrrr rs(  rN) r$Zconfiguration_utilsrutilsrZ get_loggerr!loggerZ$REGNET_PRETRAINED_CONFIG_ARCHIVE_MAPrrrrr s