Simpler approach to loading entrypoints in hubconf works properly

pull/94/head
Ross Wightman 5 years ago
parent e0685dd415
commit 6620770665

@ -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)

Loading…
Cancel
Save