bi& <SSKrSSKrSSKrSSKrSSKJr SSKJr SSKJ r SSK J r SSKJ r SSK Jr SSKJrJrJrJrJrJr SS KJr S S KJr /S Qr\"S S55r"SS\ 5rS\S\4SjrS\\\4S\\4Sjr S\S\\4Sjr!\"S\RDS9r#0r$S"S\\S\\S\#4/\S\#444Sjjr%S#S\\S\\\\S4S\\\\S4S\&\4Sjjr'S\S\S\RD44Sjr(S\S \S\RD4S!jr)g)$N)Iterable) dataclass)Enum)partial) signature) ModuleType)AnyCallableget_argsOptionalTypeVarUnion)nn)load_state_dict_from_url) WeightsEnumWeights get_modelget_model_builderget_model_weights get_weight list_modelscT\rSrSr%Sr\\S'\\S'\\\ 4\S'S\ S\ 4Sjr S r g ) ra This class is used to group important attributes associated with the pre-trained weights. Args: url (str): The location where we find the weights. transforms (Callable): A callable that constructs the preprocessing method (or validation preset transforms) needed to use the model. The reason we attach a constructor method rather than an already constructed object is because the specific object might have memory and thus we want to delay initialization until needed. meta (Dict[str, Any]): Stores meta-data related to the weights of the model and its configuration. These can be informative attributes (for example the number of parameters/flops, recipe link/methods used in training etc), configuration parameters (for example the `num_classes`) needed to construct the model or important meta-data (for example the `classes` of a classification model) needed to use the model. url transformsmetaotherreturnc|[U[5(d[$URUR:wagURUR:wag[UR [ 5(a[UR [ 5(aUR RUR R:H=(aa UR RUR R:H=(a- UR RUR R:H$UR UR :H$)NF) isinstancerNotImplementedrrrrfuncargskeywords)selfrs Q/home/james-whalen/.local/lib/python3.13/site-packages/torchvision/models/_api.py__eq__Weights.__eq__*s%))! ! 88uyy  99 " doow / /Ju?O?OQX4Y4Y$$(8(8(=(==JOO((E,<,<,A,AAJOO,,0@0@0I0II  ??e&6&66 6N) __name__ __module__ __qualname____firstlineno____doc__str__annotations__r dictr boolr(__static_attributes__r+r*r'rrs5  H sCx.7C7D7r*rc\rSrSrSr\S\S\4Sj5rS\S\S\\ \44Sjr S\ 4S jr \ S 5r \ S 5r\ S 5rS rg)rEa5 This class is the parent class of all model weights. Each model building method receives an optional `weights` parameter with its associated pre-trained weights. It inherits from `Enum` and its values should be of type `Weights`. Args: value (Weights): The data class entry with the weight information. objrcUbu[U5[La#XRURS-S5nU$[ X5(d0[ SURSUR RS35eU$)N.z(Invalid Weight class provided; expected z but received )typer1replacer,r! TypeError __class__)clsr8s r'verifyWeightsEnum.verifyOs| ?CyC++cllS&8"=>   ))>s||nN[^[h[h[q[qZrrst r*r$kwargsc4[UR/UQ70UD6$N)rr)r&r$rCs r'get_state_dictWeightsEnum.get_state_dictZs'B4B6BBr*cLURRSUR3$)Nr:)r?r,_name_r&s r'__repr__WeightsEnum.__repr__]s"..))*!DKK=99r*c.URR$rE)valuerrJs r'rWeightsEnum.url`szz~~r*c.URR$rE)rNrrJs r'rWeightsEnum.transformsdszz$$$r*c.URR$rE)rNrrJs r'rWeightsEnum.metahszzr*r+N)r,r-r.r/r0 classmethodr rAr3r1rFrKpropertyrrrr5r+r*r'rrEsCCC3C4S>C:#:%%r*rnamercURS5upSR[R[ R RS5SS5n[ R"U5nU/[R"U[R5Vs/sH-nUSRRS5(dM(USPM/ sn-nSnUH=nURRUS5n U cM$[U [ 5(dM;U n O Uc[SUS 35eXr$![a [SUS35ef=fs snf) z Gets the weights enum value by its full name. Example: "ResNet50_Weights.IMAGENET1K_V1" Args: name (str): The name of the weight enum entry. Returns: WeightsEnum: The requested weight enum. r:zInvalid weight name provided: ''.Nz __init__.pyzThe weight enum 'z0' for the specific method couldn't be retrieved.)split ValueErrorjoinsysmodulesr, importlib import_moduleinspect getmembersismodule__file__endswith__dict__get issubclassr) rV enum_name value_namebase_module_name base_modulex model_modules weights_enummpotential_classs r'rrmsCE $ 3 xx H 5 > > D DS I#2 NO))*:;K M##K1A1AB%BA Q4== ! !- 0 !B%M L **..D9  &:o{+S+S*L   ,YK7ghii  ##+ E:4&CDDE %sD*'E E*Ec\[U[5(a [U5OUn[U5$)a Returns the weights enum class associated to the given model. Args: name (callable or str): The model builder function or the name under which it is registered. Returns: weights_enum (WeightsEnum): The weights enum class associated with the model. )r!r1r_get_enum_from_fn)rVmodels r'rrs((2$'<'< d #$E U ##r*fnc[U5nSUR;a [S5eURSRnSn[ U[ 5(a[ U[5(aUnOB[U5H3n[ U[ 5(dM[ U[5(dM1Un O Uc [S5eU$)z Internal method that gets the weight enum of a specific model builder method. Args: fn (Callable): The builder method used to create the model. Returns: WeightsEnum: The requested weight enum. weightsz-The method is missing the 'weights' argument.NzjThe WeightsEnum class for the specific method couldn't be retrieved. Make sure the typing info is correct.) r parametersr\ annotationr!r<rirr )rvsigannrpts r'rtrts B-C&HII .. # . .CL#tC!=!= #A!T""z!['A'A   x   r*M)bound.cT^S[S[4S[S[44U4SjjnU$)Nrv.rcn>TbTO URnU[;a[SUS35eU[U'U$)Nz/An entry is already registered under the name 'rX)r,BUILTIN_MODELSr\)rvkeyrVs r'wrapperregister_model..wrappers@&dBKK . NseSUVW W s r*)r r~)rVrs` r'register_modelrs/HS!V$#q&)9 Nr*moduleincludeexcludec[R5VVs1sH9up4Ub/URRSS5SUR:XdM7UiM; nnnU(aN[ 5n[ U[5(aU/nUH%nU[ [R"XW55-nM' OUnU(aC[ U[5(aU/nUH%nU[ [R"XX55- nM' [U5$s snnf)aw Returns a list with the names of registered models. Args: module (ModuleType, optional): The module from which we want to extract the available models. include (str or Iterable[str], optional): Filter(s) for including the models from the set of all models. Filters are passed to `fnmatch `__ to match Unix shell-style wildcards. In case of many filters, the results is the union of individual filters. exclude (str or Iterable[str], optional): Filter(s) applied after include_filters to remove models. Filter are passed to `fnmatch `__ to match Unix shell-style wildcards. In case of many filters, the results is removal of all the models that match any individual filter. Returns: models (list): A list with the names of available models. r:rZr) ritemsr-rsplitr,setr!r1fnmatchfiltersorted) rrrkv all_modelsmodelsinclude_filterexclude_filters r'rrs*%**,,da!,,BUBUVY[\B]^_B`djdsdsBs,5 gs # #iG%Nc'.."LMMF& gs # #iG%Nc'.."LMMF& &>#s 6DDcrUR5n[UnU$![a [SU35ef=f)z Gets the model name and returns the model builder method. Args: name (str): The name under which the model is registered. Returns: fn (Callable): The model builder method. zUnknown model )lowerrKeyErrorr\)rVrvs r'rrsH ::$0112s 6configc ([U5nU"S0UD6$)a  Gets the model name and configuration and returns an instantiated model. Args: name (str): The name under which the model is registered. **config (Any): parameters passed to the model builder method. Returns: model (nn.Module): The initialized model. r+)r)rVrrvs r'rr s 4 B <<r*rE)NNN)*rr`rbr^collections.abcr dataclassesrenumr functoolsrrtypesrtypingr r r r r rtorchr_internally_replaced_utilsr__all__rrr1rr<rrtModuler~rrlistrrrr+r*r'rs $!DDA y ,7,7 ,7^%$%P!$S!$[!$H $E(C-0 $T+5F $(tK'8@ Cryy!#(HS!V