parent
e0685dd415
commit
6620770665
@ -1,10 +1,4 @@
|
|||||||
dependencies = ['torch']
|
dependencies = ['torch']
|
||||||
|
|
||||||
from timm.models import registry
|
from timm.models import registry
|
||||||
|
|
||||||
current_module = __import__(__name__)
|
globals().update(registry._model_entrypoints)
|
||||||
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)
|
|
||||||
|
|
||||||
|
Loading…
Reference in new issue