mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

timm.models: explicitly export attributes

Browse Source
This commit is contained in:
Adam J. Stewart 2025-02-21 19:33:00 +01:00 committed by Ross Wightman
parent a667d3d8f0
commit 92682d8d4d
1 changed files with 79 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
timm/models/__init__.py
View File

@ -81,20 +81,82 @@ from .xception import *
from .xception_aligned import * from .xception_aligned import *
from .xcit import * from .xcit import *
from ._builder import build_model_with_cfg, load_pretrained, load_custom_pretrained, resolve_pretrained_cfg, \ from ._builder import (
set_pretrained_download_progress, set_pretrained_check_hash build_model_with_cfg as build_model_with_cfg,
from ._factory import create_model, parse_model_name, safe_model_name load_pretrained as load_pretrained,
from ._features import FeatureInfo, FeatureHooks, FeatureHookNet, FeatureListNet, FeatureDictNet load_custom_pretrained as load_custom_pretrained,
from ._features_fx import FeatureGraphNet, GraphExtractNet, create_feature_extractor, get_graph_node_names, \ resolve_pretrained_cfg as resolve_pretrained_cfg,
register_notrace_module, is_notrace_module, get_notrace_modules, \ set_pretrained_download_progress as set_pretrained_download_progress,
register_notrace_function, is_notrace_function, get_notrace_functions set_pretrained_check_hash as set_pretrained_check_hash,
from ._helpers import clean_state_dict, load_state_dict, load_checkpoint, remap_state_dict, resume_checkpoint )
from ._hub import load_model_config_from_hf, load_state_dict_from_hf, push_to_hf_hub from ._factory import (
from ._manipulate import model_parameters, named_apply, named_modules, named_modules_with_params, \ create_model as create_model,
group_modules, group_parameters, checkpoint_seq, checkpoint, adapt_input_conv parse_model_name as parse_model_name,
from ._pretrained import PretrainedCfg, DefaultCfg, filter_pretrained_cfg safe_model_name as safe_model_name,
from ._prune import adapt_model_from_string )
from ._registry import split_model_name_tag, get_arch_name, generate_default_cfgs, register_model, \ from ._features import (
register_model_deprecations, model_entrypoint, list_models, list_pretrained, get_deprecated_models, \ FeatureInfo as FeatureInfo,
is_model, list_modules, is_model_in_modules, is_model_pretrained, get_pretrained_cfg, get_pretrained_cfg_value, \ FeatureHooks as FeatureHooks,
get_arch_pretrained_cfgs FeatureHookNet as FeatureHookNet,
FeatureListNet as FeatureListNet,
FeatureDictNet as FeatureDictNet,
)
from ._features_fx import (
FeatureGraphNet as FeatureGraphNet,
GraphExtractNet as GraphExtractNet,
create_feature_extractor as create_feature_extractor,
get_graph_node_names as get_graph_node_names,
register_notrace_module as register_notrace_module,
is_notrace_module as is_notrace_module,
get_notrace_modules as get_notrace_modules,
register_notrace_function as register_notrace_function,
is_notrace_function as is_notrace_function,
get_notrace_functions as get_notrace_functions,
)
from ._helpers import (
clean_state_dict as clean_state_dict,
load_state_dict as load_state_dict,
load_checkpoint as load_checkpoint,
remap_state_dict as remap_state_dict,
resume_checkpoint as resume_checkpoint,
)
from ._hub import (
load_model_config_from_hf as load_model_config_from_hf,
load_state_dict_from_hf as load_state_dict_from_hf,
push_to_hf_hub as push_to_hf_hub,
)
from ._manipulate import (
model_parameters as model_parameters,
named_apply as named_apply,
named_modules as named_modules,
named_modules_with_params as named_modules_with_params,
group_modules as group_modules,
group_parameters as group_parameters,
checkpoint_seq as checkpoint_seq,
checkpoint as checkpoint,
adapt_input_conv as adapt_input_conv,
)
from ._pretrained import (
PretrainedCfg as PretrainedCfg,
DefaultCfg as DefaultCfg,
filter_pretrained_cfg as filter_pretrained_cfg,
)
from ._prune import adapt_model_from_string as adapt_model_from_string
from ._registry import (
split_model_name_tag as split_model_name_tag,
get_arch_name as get_arch_name,
generate_default_cfgs as generate_default_cfgs,
register_model as register_model,
register_model_deprecations as register_model_deprecations,
model_entrypoint as model_entrypoint,
list_models as list_models,
list_pretrained as list_pretrained,
get_deprecated_models as get_deprecated_models,
is_model as is_model,
list_modules as list_modules,
is_model_in_modules as is_model_in_modules,
is_model_pretrained as is_model_pretrained,
get_pretrained_cfg as get_pretrained_cfg,
get_pretrained_cfg_value as get_pretrained_cfg_value,
get_arch_pretrained_cfgs as get_arch_pretrained_cfgs,
)