mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

timm: add __all__ to __init__

pull/2426/head
Adam J. Stewart 2025-01-11 14:31:41 +01:00 committed by Ross Wightman
parent bda46f8e6f
commit 4de1abf837
1 changed files with 28 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
timm/__init__.py
View File

@ -1,4 +1,30 @@
from .version import __version__
from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable
from .models import create_model, list_models, list_pretrained, is_model, list_modules, model_entrypoint, \
is_model_pretrained, get_pretrained_cfg, get_pretrained_cfg_value
from .models import (
create_model,
list_models,
list_pretrained,
is_model,
list_modules,
model_entrypoint,
is_model_pretrained,
get_pretrained_cfg,
get_pretrained_cfg_value,
)
__all__ = [
"__version__",
"is_scriptable",
"is_exportable",
"set_scriptable",
"set_exportable",
"create_model",
"list_models",
"list_pretrained",
"is_model",
"list_modules",
"model_entrypoint",
"is_model_pretrained",
"get_pretrained_cfg",
"get_pretrained_cfg_value",
]