You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
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)
|
|
|
