From a8ffa8379c804be8d593ba6a8d1b674feb62d79e Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Fri, 22 Nov 2019 16:03:05 +0800
Subject: [PATCH] Fix init tokens of sa nas.

---
 paddleslim/nas/search_space/combine_search_space.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddleslim/nas/search_space/combine_search_space.py b/paddleslim/nas/search_space/combine_search_space.py
index 37459ebc..17ebbd39 100644
--- a/paddleslim/nas/search_space/combine_search_space.py
+++ b/paddleslim/nas/search_space/combine_search_space.py
@@ -39,6 +39,7 @@ class CombineSearchSpace(object):
         for config_list in config_lists:
             key, config = config_list
             self.spaces.append(self._get_single_search_space(key, config))
+        self.init_tokens()
 
     def _get_single_search_space(self, key, config):
         """
@@ -56,7 +57,6 @@ class CombineSearchSpace(object):
                     config['output_size'],
                     config['block_num'],
                     block_mask=block_mask)
-
         return space
 
     def init_tokens(self):
-- 
GitLab