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): ...@@ -98,18 +98,19 @@ class MobileNetV1Space(SearchSpaceBase):
""" """
Get range table of current search space, constrains the range of tokens. Get range table of current search space, constrains the range of tokens.
""" """
base_range_table = [ # yapf: disable
len(self.head_num), len(self.filter_num1), len(self.filter_num2), base_range_table = [len(self.head_num),
len(self.k_size), len(self.filter_num2), len(self.filter_num3), len(self.filter_num1), len(self.filter_num2), len(self.k_size),
len(self.k_size), len(self.filter_num3), len(self.filter_num4), len(self.filter_num2), len(self.filter_num3), len(self.k_size),
len(self.k_size), len(self.filter_num4), len(self.filter_num5), len(self.filter_num3), len(self.filter_num4), len(self.k_size),
len(self.k_size), len(self.filter_num5), len(self.filter_num6), len(self.filter_num4), len(self.filter_num5), len(self.k_size),
len(self.k_size), len(self.filter_num6), len(self.filter_num7), len(self.filter_num5), len(self.filter_num6), len(self.k_size),
len(self.k_size), len(self.repeat), len(self.filter_num7), len(self.filter_num6), len(self.filter_num7), len(self.k_size),
len(self.filter_num8), len(self.k_size), len(self.filter_num8), len(self.repeat),
len(self.filter_num9), len(self.k_size), len(self.filter_num9), len(self.filter_num7), len(self.filter_num8), len(self.k_size),
len(self.filter_num9), 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] return base_range_table[:self.token_len]
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.
先完成此消息的编辑！
想要评论请 注册 或