pytorch-image-models/timm/__init__.py

31 lines
654 B
Python
Raw Normal View History

from .version import __version__
from .layers import is_scriptable, is_exportable, set_scriptable, set_exportable
2025-01-11 21:31:41 +08:00
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",
]