:j .ddlmZddlmZmZddlZddlmZddlm Z ddl m Z ddl m Z mZmZdd lmZdd lmZmZgd ZGd d ej,ZGddeZe edej2fddddeedededefdZy))partial)AnyOptionalN)ImageClassification)_log_api_usage_once)register_modelWeights WeightsEnum)_IMAGENET_CATEGORIES)_ovewrite_named_paramhandle_legacy_interface)AlexNetAlexNet_WeightsalexnetcfeZdZddededdffd ZdejdejfdZxZ S) r num_classesdropoutreturnNc>t|t|tjtj dddddtj dtjdd tj dd d d tj dtjdd tj d d dd tj dtj d ddd tj dtj dddd tj dtjdd  |_tjd|_ tjtj|tjddtj dtj|tjddtj dtjd||_ y)N@ r) kernel_sizestridepaddingT)inplace)rr)rrir )r#)pi$i)super__init__rnn SequentialConv2dReLU MaxPool2dfeaturesAdaptiveAvgPool2davgpoolDropoutLinear classifier)selfrr __class__s o/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/torchvision/models/alexnet.pyr&zAlexNet.__init__sf D! IIaAq A GGD ! LLQq 1 IIb#1a 8 GGD ! LLQq 1 IIc3Aq 9 GGD ! IIc3Aq 9 GGD ! IIc3Aq 9 GGD ! LLQq 1  ++F3 -- JJ ! IIk4 ( GGD ! JJ ! IIdD ! GGD ! IIdK ( xc|j|}|j|}tj|d}|j |}|S)Nr )r,r.torchflattenr1)r2r6s r4forwardzAlexNet.forward/s@ MM!  LLO MM!Q  OOA r5)ig?) __name__ __module__ __qualname__intfloatr&r8Tensorr: __classcell__)r3s@r4rrs8 C   :%,,r5rc ReZdZedeedddedddd d id d d dZeZy)rz" .m##  ! M*Gr5r pretrained)weightsT)rUprogressrUrVkwargsrc tj|}|#t|dt|jdt di|}|"|j |j|d|S)aAlexNet model architecture from `One weird trick for parallelizing convolutional neural networks `__. .. note:: AlexNet was originally introduced in the `ImageNet Classification with Deep Convolutional Neural Networks `__ paper. Our implementation is based instead on the "One weird trick" paper above. Args: weights (:class:`~torchvision.models.AlexNet_Weights`, optional): The pretrained weights to use. See :class:`~torchvision.models.AlexNet_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.AlexNet`` base class. Please refer to the `source code `_ for more details about this class. .. autoclass:: torchvision.models.AlexNet_Weights :members: rrHT)rV check_hashrS)rverifyrlenrPrload_state_dictget_state_dict)rUrVrWmodels r4rrPsk:$$W-GfmSl9S5TU  f E g44hSW4XY Lr5) functoolsrtypingrrr8torch.nnr'transforms._presetsrutilsr_apir r r _metar _utilsrr__all__ModulerrrQboolrrSr5r4rjs 5'66'B 4#bii#Lk2,0M0M!NO484%1%D%[^%cj%P%r5