提交 2be7d257 编写于 作者: W wanghaoshuang

Fix format.

上级 a7270cfb
...@@ -57,47 +57,32 @@ class MobileNetV2Space(SearchSpaceBase): ...@@ -57,47 +57,32 @@ class MobileNetV2Space(SearchSpaceBase):
each line in the following represent the index of the [expansion_factor, filter_num, repeat_num, kernel_size] each line in the following represent the index of the [expansion_factor, filter_num, repeat_num, kernel_size]
""" """
# original MobileNetV2 # original MobileNetV2
return [ # yapf: disable
4, # 1, 16, 1 return [4, # 1, 16, 1
4, 4, 5, 1, 0, # 6, 24, 1
5, 4, 5, 1, 0, # 6, 24, 2
1, 4, 4, 2, 0, # 6, 32, 3
0, # 6, 24, 1 4, 4, 3, 0, # 6, 64, 4
4, 4, 5, 2, 0, # 6, 96, 3
5, 4, 7, 2, 0, # 6, 160, 3
1, 4, 9, 0, 0] # 6, 320, 1
0, # 6, 24, 2 # yapf: enable
4,
4,
2,
0, # 6, 32, 3
4,
4,
3,
0, # 6, 64, 4
4,
5,
2,
0, # 6, 96, 3
4,
7,
2,
0, # 6, 160, 3
4,
9,
0,
0
] # 6, 320, 1
def range_table(self): def range_table(self):
""" """
get range table of current search space get range table of current search space
""" """
# head_num + 7 * [multiple(expansion_factor), filter_num, repeat, kernel_size] # head_num + 7 * [multiple(expansion_factor), filter_num, repeat, kernel_size]
return [ # yapf: disable
7, 5, 8, 6, 2, 5, 8, 6, 2, 5, 8, 6, 2, 5, 8, 6, 2, 5, 10, 6, 2, 5, return [7,
10, 6, 2, 5, 12, 6, 2 5, 8, 6, 2,
] 5, 8, 6, 2,
5, 8, 6, 2,
5, 8, 6, 2,
5, 10, 6, 2,
5, 10, 6, 2,
5, 12, 6, 2]
# yapf: enable
def token2arch(self, tokens=None): def token2arch(self, tokens=None):
""" """
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册