fix nas in python3 (PaddlePaddle#21)

* fix py3 * fix
lizexu123 · Jan 2, 2020 · cc1cd6f · cc1cd6f
1 parent 479e10d
commit cc1cd6f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/paddleslim/common/sa_controller.py b/paddleslim/common/sa_controller.py
@@ -137,7 +137,7 @@ def next_tokens(self, control_token=None):
             _logger.debug("change index[{}] from {} to {}".format(
                 index, tokens[index], new_tokens[index]))
 
-            if self._searched.has_key(str(new_tokens)):
+            if str(new_tokens) in self._searched.keys():
                 _logger.debug('get next tokens including searched tokens: {}'.
                               format(new_tokens))
                 continue

diff --git a/paddleslim/nas/sa_nas.py b/paddleslim/nas/sa_nas.py
@@ -93,7 +93,7 @@ def __init__(self,
                 premax_reward = scene['_max_reward']
                 prebest_tokens = scene['_best_tokens']
                 preiter = scene['_iter']
-                psearched = screen['_searched']
+                psearched = scene['_searched']
             else:
                 preinit_tokens = init_tokens
                 prereward = -1
@@ -114,7 +114,7 @@ def __init__(self,
                 best_tokens=prebest_tokens,
                 constrain_func=None,
                 checkpoints=save_checkpoint,
-                searched = psearched)
+                searched=psearched)
 
             max_client_num = 100
             self._controller_server = ControllerServer(