提交 a8ffa837 编写于 作者: W wanghaoshuang

Fix init tokens of sa nas.

上级 7737d044
...@@ -39,6 +39,7 @@ class CombineSearchSpace(object): ...@@ -39,6 +39,7 @@ class CombineSearchSpace(object):
for config_list in config_lists: for config_list in config_lists:
key, config = config_list key, config = config_list
self.spaces.append(self._get_single_search_space(key, config)) self.spaces.append(self._get_single_search_space(key, config))
self.init_tokens()
def _get_single_search_space(self, key, config): def _get_single_search_space(self, key, config):
""" """
...@@ -56,7 +57,6 @@ class CombineSearchSpace(object): ...@@ -56,7 +57,6 @@ class CombineSearchSpace(object):
config['output_size'], config['output_size'],
config['block_num'], config['block_num'],
block_mask=block_mask) block_mask=block_mask)
return space return space
def init_tokens(self): def init_tokens(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册