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

Simpler approach to loading entrypoints in hubconf works properly

Browse Source
This commit is contained in:
Ross Wightman 2020-02-18 14:58:23 -08:00
parent e0685dd415
commit 6620770665
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
hubconf.py
View File

@ -1,10 +1,4 @@
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)
globals().update(registry._model_entrypoints)