Skip to content

P
PaddleSlim

PaddlePaddle / PaddleSlim
接近 2 年 前同步成功

通知 51
Star 1434
Fork 344
P
PaddleSlim
提交 ffad56ea 编写于 作者: C ceci3
浏览文件
操作

update mobilenetv1

上级 18c86052
隐藏空白更改
内联 并排
Showing with 13 addition and 12 deletion
paddleslim/nas/search_space/mobilenetv1.py
浏览文件 @ ffad56ea
......@@ -98,18 +98,19 @@ class MobileNetV1Space(SearchSpaceBase):
"""
Get range table of current search space, constrains the range of tokens.
"""
base_range_table = [
len(self.head_num), len(self.filter_num1), len(self.filter_num2),
len(self.k_size), len(self.filter_num2), len(self.filter_num3),
len(self.k_size), len(self.filter_num3), len(self.filter_num4),
len(self.k_size), len(self.filter_num4), len(self.filter_num5),
len(self.k_size), len(self.filter_num5), len(self.filter_num6),
len(self.k_size), len(self.filter_num6), len(self.filter_num7),
len(self.k_size), len(self.repeat), len(self.filter_num7),
len(self.filter_num8), len(self.k_size), len(self.filter_num8),
len(self.filter_num9), len(self.k_size), len(self.filter_num9),
len(self.filter_num9), len(self.k_size)
]
# yapf: disable
base_range_table = [len(self.head_num),
len(self.filter_num1), len(self.filter_num2), len(self.k_size),
len(self.filter_num2), len(self.filter_num3), len(self.k_size),
len(self.filter_num3), len(self.filter_num4), len(self.k_size),
len(self.filter_num4), len(self.filter_num5), len(self.k_size),
len(self.filter_num5), len(self.filter_num6), len(self.k_size),
len(self.filter_num6), len(self.filter_num7), len(self.k_size),
len(self.repeat),
len(self.filter_num7), len(self.filter_num8), len(self.k_size),
len(self.filter_num8), len(self.filter_num9), len(self.k_size),
len(self.filter_num9), len(self.filter_num9), len(self.k_size)]
# yapf: enable
return base_range_table[:self.token_len]
def token2arch(self, tokens=None):
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或