From fa15665cde8ac91acd5a6deec0cf4b281dac751d Mon Sep 17 00:00:00 2001
From: ceci3 <592712189@qq.com>
Date: Wed, 4 Dec 2019 03:51:13 +0000
Subject: [PATCH] fix

---
 paddleslim/nas/sa_nas.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddleslim/nas/sa_nas.py b/paddleslim/nas/sa_nas.py
index 1a284cd4..38cdb186 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(
-- 
GitLab