pretrained-models.pytorch
pretrained-models.pytorch copied to clipboard
Update __init__.py
Best practices to import with "*", also fixes my use case of automatically generate lookup dictionary by having the names of the models being exported:
from pretrainedmodels.models import __all__ as pretrained_models_names
pretrained_models_lookup = {
name: getattr(importlib.import_module("pretrainedmodels.models"), name)
for name in pretrained_models_names
}
lookup = {**pretrained_models_lookup}
Thanks! I'll look into it asap