diff --git a/paddleslim/nas/sa_nas.py b/paddleslim/nas/sa_nas.py
index 1a284cd45c75971f1701e3b8638696d7894db55d..38cdb186bf108d36a754c9c5de9444b82f65368d 100644
--- a/paddleslim/nas/sa_nas.py
+++ b/paddleslim/nas/sa_nas.py
@@ -90,10 +90,10 @@ class SANAS(object):
                 prebest_tokens = scene['_best_tokens']
                 preiter = scene['_iter']
             else:
-                preinit_tokens = None
+                preinit_tokens = init_tokens
                 prereward = -1
                 premax_reward = -1
-                prebest_tokens = init_tokens
+                prebest_tokens = None
                 preiter = 0
                       
             controller = SAController(