o )i;" @sddlmZddlmZmZddlZddlmZddlmm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZmZgd ZGd d d ejZGdddejZdedeedededef ddZedddZGdddeZGdddeZ eedej!fddd d!deedededefd"d#Z"eede j!fddd d!dee dededefd$d%Z#dS)&)partial)AnyOptionalN)ImageClassification)_log_api_usage_once)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface) SqueezeNetSqueezeNet1_0_WeightsSqueezeNet1_1_Weights squeezenet1_0 squeezenet1_1c sHeZdZdededededdf fdd Zd ejdejfd d ZZS) Fireinplanessqueeze_planesexpand1x1_planesexpand3x3_planesreturnNcsvt||_tj||dd|_tjdd|_tj||dd|_tjdd|_ tj||ddd|_ tjdd|_ dS)Nr kernel_sizeTZinplace)rpadding) super__init__rnnConv2dsqueezeReLUsqueeze_activation expand1x1expand1x1_activation expand3x3expand3x3_activation)selfrrrr __class__l/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torchvision/models/squeezenet.pyr s z Fire.__init__xcCs8|||}t||||||gdSNr)r%r#torchcatr'r&r)r(r*r/r-r-r.forwards z Fire.forward) __name__ __module__ __qualname__intr r1Tensorr4 __classcell__r-r-r+r.rs" rc sFeZdZddedededdffd d Zd ejdejfd d Z Z S)r1_0?version num_classesdropoutrNcs6tt|||_|dkrhttjdddddtjddtjdddd t dd d d t d d d d t d d d d tjdddd t dd d d t ddddt ddddt dd ddtjdddd t dd dd |_ nd|dkrttjdd dddtjddtjdddd t d d d d t d d d d tjdddd t d d d d t dd d d tjdddd t ddddt ddddt dd ddt dd dd |_ nt d|dtjd|jdd}ttj |d|tjddt d|_|D]+}t|tjr||urtj|jdddnt|j|jdurt|jdqdS)Nr;r`r)rstrideTr)rrCZ ceil_mode@ 0ii1_1zUnsupported SqueezeNet version z: 1_0 or 1_1 expectedrr)p)rrgg{Gz?)meanZstdr)rr rr?r!Z Sequentialr"r$Z MaxPool2drfeatures ValueErrorZDropoutZAdaptiveAvgPool2d classifiermodules isinstanceinitZnormal_weightZkaiming_uniform_ZbiasZ constant_)r*r>r?r@Z final_convmr+r-r.r %sb                       zSqueezeNet.__init__r/cCs ||}||}t|dSr0)rNrPr1flattenr3r-r-r.r4^s   zSqueezeNet.forward)r;r<r=) r5r6r7strr8floatr r1r9r4r:r-r-r+r.r$s 9rr>weightsprogresskwargsrcKsN|durt|dt|jdt|fi|}|dur%||j|dd|S)Nr? categoriesT)rZZ check_hash)r lenmetarZload_state_dictZget_state_dict)r>rYrZr[modelr-r-r. _squeezenetds r`z@https://github.com/pytorch/vision/pull/49#issuecomment-277560717zXThese weights reproduce closely the results of the paper using a simple training recipe.)r\ZrecipeZ_docsc @DeZdZedeeddiedddddd id d d d ZeZdS)rz>https://download.pytorch.org/models/squeezenet1_0-b66bff10.pthZ crop_size)rdi  ImageNet-1Kg M@g{GT@zacc@1zacc@5gh|?5?g&1@Zmin_sizeZ num_paramsZ_metricsZ_ops _file_sizeurlZ transformsr^N r5r6r7r rr _COMMON_META IMAGENET1K_V1DEFAULTr-r-r-r.r|& rc @ra)rz>https://download.pytorch.org/models/squeezenet1_1-b8a52dc0.pthrbrc)rpi(regX9M@g-'T@rfgtV?g"~@rgriNrkr-r-r-r.rrorZ pretrained)rYT)rYrZcKt|}td||fi|S)aSqueezeNet model architecture from the `SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size `_ paper. Args: weights (:class:`~torchvision.models.SqueezeNet1_0_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.SqueezeNet1_0_Weights` below for more details, and possible values. By default, no pre-trained weights are used. progress (bool, optional): If True, displays a progress bar of the download to stderr. Default is True. **kwargs: parameters passed to the ``torchvision.models.squeezenet.SqueezeNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.SqueezeNet1_0_Weights :members: r;)rverifyr`rYrZr[r-r-r.rs rcKrq)a/SqueezeNet 1.1 model from the `official SqueezeNet repo `_. SqueezeNet 1.1 has 2.4x less computation and slightly fewer parameters than SqueezeNet 1.0, without sacrificing accuracy. Args: weights (:class:`~torchvision.models.SqueezeNet1_1_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.SqueezeNet1_1_Weights` below for more details, and possible values. By default, no pre-trained weights are used. progress (bool, optional): If True, displays a progress bar of the download to stderr. Default is True. **kwargs: parameters passed to the ``torchvision.models.squeezenet.SqueezeNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.SqueezeNet1_1_Weights :members: rK)rrrr`rsr-r-r.rs r)$ functoolsrtypingrrr1Ztorch.nnr!Z torch.nn.initrSZtransforms._presetsrutilsrZ_apir r r _metar _utilsr r__all__ModulerrrWboolr`rlrrrmrrr-r-r-r.sl     @