diff --git a/timm/models/__init__.py b/timm/models/__init__.py index 4ef966ea..7119c4f5 100644 --- a/timm/models/__init__.py +++ b/timm/models/__init__.py @@ -7,7 +7,7 @@ from .senet import * from .xception import * from .nasnet import * from .pnasnet import * -from .gen_efficientnet import * +from .efficientnet import * from .mobilenetv3 import * from .inception_v3 import * from .gluon_resnet import * diff --git a/timm/models/gen_efficientnet.py b/timm/models/efficientnet.py similarity index 99% rename from timm/models/gen_efficientnet.py rename to timm/models/efficientnet.py index fe20ff13..2ed2a14e 100644 --- a/timm/models/gen_efficientnet.py +++ b/timm/models/efficientnet.py @@ -1294,6 +1294,3 @@ def tf_mixnet_l(pretrained=False, **kwargs): 'tf_mixnet_l', channel_multiplier=1.3, pretrained=pretrained, **kwargs) return model - -def gen_efficientnet_model_names(): - return set(_models)