From 6d5f438b65b657b4bb943a533ed6a00ef123c6e9 Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Fri, 8 Nov 2019 13:09:40 +0800
Subject: [PATCH] rename mobilenetv2_space to mobilenetv2

---
 paddleslim/nas/search_space/__init__.py                     | 6 ++++--
 .../search_space/{mobilenetv2_space.py => mobilenetv2.py}   | 2 ++
 2 files changed, 6 insertions(+), 2 deletions(-)
 rename paddleslim/nas/search_space/{mobilenetv2_space.py => mobilenetv2.py} (99%)

diff --git a/paddleslim/nas/search_space/__init__.py b/paddleslim/nas/search_space/__init__.py
index 767760fa..0e9157d9 100644
--- a/paddleslim/nas/search_space/__init__.py
+++ b/paddleslim/nas/search_space/__init__.py
@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from .mobilenetv2_space import MobileNetV2Space
+import mobilenetv2
+from .mobilenetv2 import *
 import search_space_registry
 from search_space_registry import *
 import search_space_factory
@@ -20,7 +21,8 @@ from search_space_factory import *
 import search_space_base
 from search_space_base import *
 
-__all__ = ["MobileNetV2Space"]
+__all__ = []
+__all__ += mobilenetv2.__all__
 __all__ += search_space_registry.__all__
 __all__ += search_space_factory.__all__
 __all__ += search_space_base.__all__
diff --git a/paddleslim/nas/search_space/mobilenetv2_space.py b/paddleslim/nas/search_space/mobilenetv2.py
similarity index 99%
rename from paddleslim/nas/search_space/mobilenetv2_space.py
rename to paddleslim/nas/search_space/mobilenetv2.py
index 4cd516ff..270da2b1 100644
--- a/paddleslim/nas/search_space/mobilenetv2_space.py
+++ b/paddleslim/nas/search_space/mobilenetv2.py
@@ -23,6 +23,8 @@ from .search_space_base import SearchSpaceBase
 from .base_layer import conv_bn_layer
 from .search_space_registry import SEARCHSPACE
 
+__all__ = ["MobileNetV2Space"]
+
 
 @SEARCHSPACE.register
 class MobileNetV2Space(SearchSpaceBase):
-- 
GitLab