diff --git a/paddleslim/nas/sa_nas.py b/paddleslim/nas/sa_nas.py
index 428385f54df579bef29b0be893e6272ef342fdda..73a8ecf6cf45ee7f32503dd204a3f739ca01cde6 100644
--- a/paddleslim/nas/sa_nas.py
+++ b/paddleslim/nas/sa_nas.py
@@ -93,7 +93,7 @@ class SANAS(object):
                 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 @@ class SANAS(object):
                 best_tokens=prebest_tokens,
                 constrain_func=None,
                 checkpoints=save_checkpoint,
-                searched = psearched)
+                searched=psearched)
 
             max_client_num = 100
             self._controller_server = ControllerServer(