mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
11 lines
286 B
Python
11 lines
286 B
Python
dependencies = ['torch']
|
|
|
|
from timm.models import registry
|
|
|
|
current_module = __import__(__name__)
|
|
current_module.__dict__.update(registry._model_entrypoints)
|
|
#for fn_name in registry.list_models():
|
|
# fn = registry.model_entrypoint(fn_name)
|
|
# setattr(current_module, fn_name, fn)
|
|
|