From 902d32fb1693a1e0756e5dc7b088e06ee0ddcb4b Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 28 Nov 2019 16:55:36 -0800 Subject: [PATCH] Renamed gen_efficientnet.py -> efficientnet.py --- timm/models/__init__.py | 2 +- timm/models/{gen_efficientnet.py => efficientnet.py} | 3 --- 2 files changed, 1 insertion(+), 4 deletions(-) rename timm/models/{gen_efficientnet.py => efficientnet.py} (99%) 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)