U
    ¯Ê¹jÚ  ã                   @   s¤  d dl T d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dl	T d dl
T d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dlT d dl T d dl!T d dl"T d dl#T d dl$T d dl%T d dl&T d dl'T d dl(T d dl)T d dl*T d dl+T d dl,T d dl-T d dl.T d dl/T d dl0T d dl1T d dl2T d dl3T d dl4T d dl5T d dl6T d dl7T d dl8T d dl9T d dl:T d dl;T d dl<T d dl=T d dl>T d dl?T d dl@T d dlAT d dlBT d dlCT d dlDT d dlET d dlFT d dlGT d dlHT d dlIT d dlJT d dlKT d dlLT d dlMmNZNmOZOmPZPmQZQmRZRmSZS d dlTmUZUmVZVmWZW d dlXmYZYmZZZm[Z[m\Z\m]Z] d dl^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZg d dlhmiZimjZjmkZkmlZlmmZm d dlnmoZompZpmqZq d dlrmsZsmtZtmuZumvZvmwZwmxZxmyZymzZz d d	l{m|Z|m}Z}m~Z~ d d
lm€Z€ d dl�m‚Z‚mƒZƒm„Z„m…Z…m†Z†m‡Z‡mˆZˆm‰Z‰mŠZŠm‹Z‹mŒZŒm�Z�mŽZŽm�Z�m�Z� dS )é   )Ú*)Úbuild_model_with_cfgÚload_pretrainedÚload_custom_pretrainedÚresolve_pretrained_cfgÚ set_pretrained_download_progressÚset_pretrained_check_hash)Úcreate_modelÚparse_model_nameÚsafe_model_name)ÚFeatureInfoÚFeatureHooksÚFeatureHookNetÚFeatureListNetÚFeatureDictNet)	ÚFeatureGraphNetÚGraphExtractNetÚcreate_feature_extractorÚregister_notrace_moduleÚis_notrace_moduleÚget_notrace_modulesÚregister_notrace_functionÚis_notrace_functionÚget_notrace_functions)Úclean_state_dictÚload_state_dictÚload_checkpointÚremap_state_dictÚresume_checkpoint)Úload_model_config_from_hfÚload_state_dict_from_hfÚpush_to_hf_hub)Úmodel_parametersÚnamed_applyÚnamed_modulesÚnamed_modules_with_paramsÚgroup_modulesÚgroup_parametersÚcheckpoint_seqÚadapt_input_conv)ÚPretrainedCfgÚ
DefaultCfgÚfilter_pretrained_cfg)Úadapt_model_from_string)Úsplit_model_name_tagÚget_arch_nameÚgenerate_default_cfgsÚregister_modelÚregister_model_deprecationsÚmodel_entrypointÚlist_modelsÚlist_pretrainedÚget_deprecated_modelsÚis_modelÚlist_modulesÚis_model_in_modulesÚis_model_pretrainedÚget_pretrained_cfgÚget_pretrained_cfg_valueN)‘ZbeitZbyoanetZbyobnetZcaitÚcoatZconvitZ	convmixerZconvnextZcrossvitZcspnetZdavitZdeitZdensenetZdlaZdpnZedgenextZefficientformerZefficientformer_v2ZefficientnetZefficientvit_mitZefficientvit_msraZevaZfastvitZfocalnetZgcvitZghostnetZhardcorenasZhgnetZhrnetZinception_nextZinception_resnet_v2Zinception_v3Zinception_v4ZlevitZmaxxvitZ
metaformerZ	mlp_mixerZmobilenetv3Z	mobilevitZmvitv2ZnasnetÚnestZnextvitZnfnetZpitZpnasnetZpvt_v2ZregnetZrepghostZrepvitZres2netZresnestZresnetZresnetv2ZrexnetZselecslsZsenetZ	sequencerZsknetZswin_transformerZswin_transformer_v2Zswin_transformer_v2_crZtiny_vitÚtntZtresnetZtwinsZvggZ	visformerZvision_transformerZvision_transformer_hybridZvision_transformer_relposZvision_transformer_samZvoloZvovnetZxceptionZxception_alignedZxcitZ_builderr   r   r   r   r   r   Ú_factoryr	   r
   r   Z	_featuresr   r   r   r   r   Z_features_fxr   r   r   r   r   r   r   r   r   Z_helpersr   r   r   r   r   Z_hubr   r    r!   Z_manipulater"   r#   r$   r%   r&   r'   r(   r)   Z_pretrainedr*   r+   r,   Z_pruner-   Ú	_registryr.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   © rB   rB   úS/var/www/html/bwave/python/venv/lib/python3.8/site-packages/timm/models/__init__.pyÚ<module>   s¬    ,(