+ j&j^RIt^RIt^RIt^RIt^RIHt^RIHt^RIH t ^RI H t ^RIH t ^RI Ht^RIHtHtHtHtHtHt^R IHt^R IHt.ROt]!R R 44t!RR ] 4tRRltRRltRRlt]!R]P@R7t!/t"R RRllt#R!RRllt$RRlt%RRlt&R#)"N)Iterable) dataclass)Enum)partial) signature) ModuleType)AnyCallableget_argsOptionalTypeVarUnion)nn)load_state_dict_from_url WeightsEnumWeightscBa]tRt^toRtV3RlRltV3RltRtVtR#)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. c&<V^8dQhRS[RS[/#)otherreturn)r bool)format __classdict__s"X/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchvision/models/_api.py __annotate__Weights.__annotate__*s77C7D7c\V\4'g\#VPVP8wdR#VPVP8wdR#\VP \ 4'd\VP \ 4'dVP PVP P8H;'dcVP PVP P8H;'d.VP PVP P8H#VP VP 8H#)F) isinstancerNotImplementedurlmeta transformsrfuncargskeywords)selfrs&&r__eq__Weights.__eq__*s%))! ! 88uyy  99 " doow / /Ju?O?OQX4Y4Y$$(8(8(=(==JJOO((E,<,<,A,AAJJOO,,0@0@0I0II  ??e&6&66 6rcT<V^8dQh/S[;R&S[;R&S[S[S[3,;R&#)rr"r$r#)strr dictr )rrs"rrrs3" H#$%& sCx.'rN) __name__ __module__ __qualname____firstlineno____doc__r)__annotate_func____static_attributes____classdictcell__rs@rrrs &77+rca]tRt^EtoRt]V3RlRl4tV3RlRltV3RlRlt] R4t ] R 4t ] R 4t R t VtR #) ra5 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. c&<V^8dQhRS[RS[/#)robjr)r )rrs"rrWeightsEnum.__annotate__Psrc Ve\V4\Jd,WPVPR,R4,pV#\ W4'g1\ RVP RVP P R24hV#)N.z(Invalid Weight class provided; expected z but received )typer,replacer/r TypeError __class__)clsr:s&&rverifyWeightsEnum.verifyOs| ?CyC++cllS&8"=>   ))>s||nN[^[h[h[q[qZrrst rcB<V^8dQhRS[RS[RS[S[S[3,/#)rr&kwargsr)r r-r,)rrs"rrr;Zs/CCCC3C4S>Crc4\VP.VO5/VB#N)rr")r(r&rGs&*,rget_state_dictWeightsEnum.get_state_dictZs'B4B6BBrc <V^8dQhRS[/#)rr)r,)rrs"rrr;]s::#:rcLVPP RVP 2#)r=)rBr/_name_r(s&r__repr__WeightsEnum.__repr__]s"..))*!DKK=99rc.VPP#rI)valuer"rOs&rr"WeightsEnum.url`szz~~rc.VPP#rI)rSr$rOs&rr$WeightsEnum.transformsdszz$$$rc.VPP#rI)rSr#rOs&rr#WeightsEnum.metahszzrr.N)r/r0r1r2r3 classmethodrDrJrPpropertyr"r$r#r5r6r7s@rrrEspCC::%%rc0V^8dQhR\R\/#rnamer)r,r)rs"rrrms!$!$S!$[!$rcVPR4wrRP\P\ ,P PR4RR4p\ P!T4pT.\P!T\P4Uu.uF6pT^,PPR4'gK-T^,NK8 up,pRpTF?pTPPTR4p T fK%\T \ 4'gK=T pM Tf\RT R24hYr,# \d\RT R24hi;iuupi) 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) r] enum_name value_namebase_module_name base_modulex model_modules weights_enummpotential_classs & r get_weightrymsEE $ 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 %sE*E EEchV^8dQhR\\\3,R\\,/#r\)rr r,r?r)rs"rrrs( $ $E(C-0 $T+5F $rc^\V\4'd \V4MTp\V4#)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 r,get_model_builder_get_enum_from_fn)r]models& rget_model_weightsrs((2$'<'< d #$E U ##rcFV^8dQhR\R\\,/#)rfnr)r r?r)rs"rrrs(tK'8rc\V4pRVP9d \R4hVPR,PpRp\ V\ 4'd\ V\4'dTpMD\V4F5p\ V\ 4'gK\ V\4'gK3TpM Vf \R4hV#)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 parametersrb annotationr r?rorr )rsigannrvts& rr}r}s B-C&HII .. # . .CL#tC!=!= #A!T""z!['A'A   x   rM)boundcV^8dQhR\\,R\\R\3,.\R\3,3,/#rr]r.)r r,r r)rs"rrrs@#(HS!V.__annotate__s,HS!V$#q&)9rcr<SeSM VPpV\9d\RV R24hV\V&V#)Nz/An entry is already registered under the name 'r_)r/BUILTIN_MODELSrb)rkeyr]s& rwrapperregister_model..wrappers@&dBKK . NseSUVW W s rr.)r]rsf rregister_modelrs Nrc V^8dQhR\\,R\\\,\R3,R\\\,\R3,R\ \,/#)rmoduleincludeNexcluder)r rrrr,list)rs"rrrsY%% Z % 8C=#t+ ,%8C=#t+ ,% #Y %rc<\P4UUu0uF?wr4Ve5VPPR^4^,VP8XgK=VkKA pppV'dT\ 4p\ V\4'dV.pVF)pV\ \P!WW44,pK+ MTpV'dI\ V\4'dV.pVF)pV\ \P!WX44, pK+ \V4#uuppi)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=) ritemsr0rsplitr/setr r,fnmatchfiltersorted) rrrkv all_modelsmodelsinclude_filterexclude_filters &&& r list_modelsrs*%**,,da!,,BUBUVY[\B]^_B`djdsdsBs,5 gs # #iG%Nc'.."LMMF& gs # #iG%Nc'.."LMMF& &>#s :DDc^V^8dQhR\R\R\P3,/#r)r,r rModule)rs"rrrs'CHS"))^$<rc|VP4p\V,pV# \d\RT 24hi;i)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 )lowerrKeyErrorrb)r]rs& rr|r|sH ::$0112s !;cPV^8dQhR\R\R\P/#)rr]configr)r,r rr)rs"rrr s%  C 3 299 rc (\V4pV!R/VB#)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|)r]rrs&, r get_modelr s 4 B <<r)rrrr|rryrrI)NNN)'rrfrhrdcollections.abcr dataclassesrenumr functoolsrrtypesrtypingr r r r r rtorchr_internally_replaced_utilsr__all__rrryrr}rrrrrr|rr.rrrs $!DDA y ,7,7 ,7^%$%P!$H $@ Cryy!%P$ r