NgOddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZmZmZmZddlmZddlmZdd lmZmZmZmZmZdd lm Z dd l!m"Z"dd l#m$Z$dd l%m&Z&ej'e(Z)da*da+e,ej-.dddkZ/gdZ0dZ1d/dZ2d/dZ3 d0de j4de ede efdZ5 d1de j4de ede,de,de ede6f dZ7d Z8d!Z9d"Z: d0d#e;d$e"fd%Z< d2d&ed#e;d'e6de ed(e ed)e ed*e ed+e6d,e ed-e e e;fd.Z=dS)3N)deepcopy)AnyCallableDictListOptionalTuple)nn)load_state_dict_from_url)FeatureListNetFeatureDictNetFeatureHookNetFeatureGetterNet)FeatureGraphNet)load_state_dict) has_hf_hubdownload_cached_filecheck_cached_fileload_state_dict_from_hfload_custom_from_hf)adapt_input_conv) PretrainedCfg)adapt_model_from_file)get_pretrained_cfgFTIMM_USE_OLD_CACHE) set_pretrained_download_progressset_pretrained_check_hashload_custom_pretrainedload_pretrainedpretrained_cfg_for_featuresresolve_pretrained_cfgbuild_model_with_cfgc8|dd}|dd}|dd}|dd}|dd}d}d}|dkrtd r d}|sJ|}n`|rd}|}t|tsJnB|rd}|}n;d }tr|rt |nd }|s|rtd rd}|}n|rd}|}|dkr |d dr ||d f}||fS) Nsourceurlfile state_dict hf_hub_idhf-hubT) necessaryFhf_hub_filename)getr isinstancedict_USE_OLD_CACHEr) pretrained_cfg cfg_sourcepretrained_urlpretrained_file pretrained_sdr) load_frompretrained_locold_cache_valids P/var/www/html/ai-engine/env/lib/python3.11/site-packages/timm/models/_builder.py_resolve_pretrained_sourcer: s##Hb11J#''t44N$((66O"&&|T::M"";55IINX*t"<"<"< "  0$I*Nnd33 3 3 3 3  0I,NN#O aGU"`"3N"C"C"C[`" 0y 0Z$5O5O5O 0$ !* 0! !/H!3!34Et!L!L'8I)JJ n $$Tc |adS)zA Set download progress for pretrained weights on/off (globally). N)_DOWNLOAD_PROGRESSenables r9rrMs r;c |adS)z= Set hash checking for pretrained weights on/off (globally). N) _CHECK_HASHr>s r9rrSs KKKr;modelr1load_fnc|pt|dd}|stddSt|\}}|stddS|dkrtdn"|dkrt |t t }||||dSt|d r||dStd dS) aLoads a custom (read non .pth) weight file Downloads checkpoint file into cache-dir like torch.hub based loaders, but calls a passed in custom load fun, or the `load_pretrained` model member fn. If the object is already present in `model_dir`, it's deserialized and returned. The default value of `model_dir` is ``/checkpoints`` where `hub_dir` is the directory returned by :func:`~torch.hub.get_dir`. Args: model: The instantiated model to load weights into pretrained_cfg (dict): Default pretrained model cfg load_fn: An external standalone fn that loads weights into provided model, otherwise a fn named 'laod_pretrained' on the model will be called if it exists r1Nz/Invalid pretrained config, cannot load weights.zHNo pretrained weights exist for this model. Using random initialization.r*zKHugging Face hub not currently supported for custom load pretrained models.r&) check_hashprogressrzXValid function to load pretrained weights is not available, using random initialization.) getattr_loggerwarningr:rrAr=hasattrr)rBr1rCr6r7s r9rrYs($Mwu6F'M'MN IJJJ :> J JI~ bcccHeffff e  - "'    ~&&&&& ) * *t n-----rsssssr; num_classesin_chans filter_fnstrictc |pt|dd}|stdt|\}}|dkrtd|}n|dkr\td|d|d d r||dSt|}n|d krtd |d|d d r3t|tt }||dS t|dttd}n#t$r t|dtt}YnwxYw|dkrtd|dt|ttfrK|d d } t| t r| dkrt#g||RdSt%|}n;t%|d}n)|dd} td| d|0 |||}n"#t$r} ||}Yd} ~ nd} ~ wwxYw|dd} | |dkrt| t r| f} | D]|} | dz} t'|||||<td| d|dC#t($r-} ||=d }td | d!Yd} ~ ud} ~ wwxYw|d"d}|d#d$}|t|t r|f}||d%kr:|D]4}||dzd||d&zd5d }nA|d$kr;|D]8}||dz}||d||dz<||d&z}||d||d&z<9|||'}|jr6td(d)|jd*|jr8td+d)|jd,dSdS)-a Load pretrained checkpoint Args: model (nn.Module) : PyTorch model module pretrained_cfg (Optional[Dict]): configuration for pretrained weights / target dataset num_classes (int): num_classes for target model in_chans (int): in_chans for target model filter_fn (Optional[Callable]): state_dict filter fn for load (takes state_dict, model as args) strict (bool): strict load of checkpoint r1NzWInvalid pretrained config, cannot load weights. Use `pretrained=False` for random init.r(z*Loading pretrained weights from state dictr'z&Loading pretrained weights from file () custom_loadFr&z%Loading pretrained weights from url ()rFrEcpuT) map_locationrFrE weights_only)rUrFrEr*z2Loading pretrained weights from Hugging Face hub (hf)rV architecturez this modelz No pretrained weights exist for z). Use `pretrained=False` for random init. first_convrLz.weightzConverted input conv z pretrained weights from 3 to z channel(s)zUnable to convert pretrained z+ weights, using random init for this layer. classifier label_offsetrrMz.bias)rPzMissing keys (z, zZ) discovered while loading pretrained weights. This is expected if model is being adapted.zUnexpected keys (zY) found while loading pretrained weights. This may be expected if model is being adapted.)rG RuntimeErrorr:rHinfor-rrrr=rAr TypeErrorr.listtuplestrrrrNotImplementedErrorrIpop missing_keysjoinunexpected_keys)rBr1rMrNrOrPr6r7r(rS model_namee input_convsinput_conv_name weight_name classifiersr[classifier_nameclassifier_weightclassifier_bias load_results r9rrs&$Mwu6F'M'MN vtuuu :> J JI~L   BCCC# f   OnOOOPPP   mU 3 3 9  ! !. 1 1 1 F(88JJ e   N^NNNOOO   mU 3 3 1+&N  ! !. 1 1 1 F 5"!&/*!%     5"!&/*    h   [.[[[\\\ ntUm 4 4 T(,,]EBBK+s++ F t0C0C#;^;U;;;;4nE 0dSSSJJ#'' EE sjsssttt /":u55JJ / / /":..JJJJJJ /!$$\488K8q== k3 ' ' )&.K* r rO)I5K r*:8ZP[E\*]*] ;' pOpp[cppprrrr& r r r{+pOppprrrrrrrr r !$$\488K!%%na88L k3 ' ' )&.K .7 7 7#. @ @:DAAA8$????FF A  #. W W$./J$K!:KLMM:Z ?Y67",_w-F"G8G 8V ?W455'' 6'BBK=  >> F " ")/M)B)BK & F " ")/M)B)BK & ##F## 22i2622)E,E 6%eW55#+nQQ}fjjYfhlNmNm0n0n  ).ZZ A..*$     !1 K ' '%//-88K+s++ I)//11 &>>>OO$8$???[(("0KK F**"0KK F**"0KK D(("1KK H,,!%J"2KKH"H;"H"HHHH' I*)KUL$77  !* !  " "< < < < E11[11:>JJ!0 Lr;)T)NN)NrKrLNT)NNNNTNN)>rloggingoscopyrtypingrrrrrr torchr torch.hubr timm.models._featuresr r rrtimm.models._features_fxrtimm.models._helpersrtimm.models._hubrrrrrtimm.models._manipulatertimm.models._pretrainedrtimm.models._prunertimm.models._registryr getLogger__name__rHr=rAintenvironr-r0__all__r:rrModulerboolrr ryrrar!r"rr;r9rs ================......bbbbbbbbbbbb444444000000444444111111444444444444 ' H % % RZ^^$8!<<==A \ \ \*%*%*%Z    *.&*+t+ty+t +t(#+t+t+t+t`*.(, }A}Ay}A }A}A }A H% }A  }A}A}A}A@&0&0&0V#       N*.15#'&*"&37.2uuuuu! u !) u C= ud^u u'x0u c +uuuuuur;