QЦiF' DSSKrSSKrSSKrSSKrSSKJr SSKJr SSKJ r SSKJ r SSK J r SSK JrJrJrJrJrJrJrJrJrJrJr SSKJr S S KJr /S Qr\"S S 55r"SS\5rS\ S\4Sjr!S\\\ 4S\\4Sjr"S\S\\4Sjr#\"S\RHS9r%0r&S!S\\ S\\S\%4/\S\%444Sjjr'S"S\\ S\\\ \ S4S\\\ \ S4S\\ 4Sjjr(S\ S\S\RH44Sjr)S\ S\S\RH4S jr*g)#N) dataclass)Enum)partial) signature) ModuleType) AnyCallableDictIterableListMappingOptionalSetTypeTypeVarUnion)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) isinstancerNotImplementedrr!r rfuncargskeywords)selfr"s V/var/www/html/ai-image-ml/venv/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 r rboolr,__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)rDa5 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. objr#cUbu[U5[La#XRURS-S5nU$[ X5(d0[ SURSUR RS35eU$)N.z(Invalid Weight class provided; expected z but received )typer5replacer0r% TypeError __class__)clsr;s r+verifyWeightsEnum.verifyNs| ?CyC++cllS&8"=>   ))>s||nN[^[h[h[q[qZrrst r.r(kwargsc4[UR/UQ70UD6$N)rr)r*r(rFs r+get_state_dictWeightsEnum.get_state_dictYs'B4B6BBr.cLURRSUR3$)Nr=)rBr0_name_r*s r+__repr__WeightsEnum.__repr__\s"..))*!DKK=99r.c.URR$rH)valuerrMs r+rWeightsEnum.url_szz~~r.c.URR$rH)rQr rMs r+r WeightsEnum.transformscszz$$$r.c.URR$rH)rQr!rMs r+r!WeightsEnum.metagszzr.r/N)r0r1r2r3r4 classmethodrrDr r5rIrNpropertyrr r!r8r/r.r+rrDsCCC3C738;LC:#:%%r.rnamer#cURS5upSR[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 ValueErrorjoinsysmodulesr0 importlib import_moduleinspect getmembersismodule__file__endswith__dict__get issubclassr) rY enum_name value_namebase_module_name base_modulex model_modules weights_enummpotential_classs r+rrlsCE $ 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%r5r_get_enum_from_fn)rYmodels r+rrs((2$'<'< d #$E U ##r.fnc[U5nSUR;a [S5e[U5RSRnSn[ U[ 5(a[ U[5(aUnOCURH3n[ 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?rlr__args__)rysigannrsts r+rwrws B-C&HII B- " "9 - 8 8CL#tC!=!= A!T""z!['A'A   x   r.M)bound.cT^S[S[4S[S[44U4SjjnU$)Nry.r#cn>TbTO URnU[;a[SUS35eU[U'U$)Nz/An entry is already registered under the name 'r[)r0BUILTIN_MODELSr_)rykeyrYs r+wrapperregister_model..wrappers@&dBKK . NseSUVW W s r.)r r)rYrs` 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=r]r) ritemsr1rsplitr0setr%r5fnmatchfiltersorted) 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_)rYrys 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)rYrrys r+rr s 4 B <<r.rH)NNN)+rrcrera dataclassesrenumr functoolsrrtypesrtypingrr r r r r rrrrrtorchr_internally_replaced_utilsr__all__rrr5rrrwModulerrrrrrr/r.r+rs !ddddA y ,7,7 ,7^%$%P!$S!$[!$H $E(C-0 $T+5F $(tK'8B Cryy!#(HS!V