From 7b33f4d6ebe0332d482d5b575d84d63e634d7eb1 Mon Sep 17 00:00:00 2001 From: ceci3 <592712189@qq.com> Date: Tue, 31 Dec 2019 08:45:35 +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 428385f5..73a8ecf6 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( -- GitLab