From 7737d04460855110554d839cb96eb35e0531c3d9 Mon Sep 17 00:00:00 2001 From: wanghaoshuang Date: Fri, 22 Nov 2019 15:50:31 +0800 Subject: [PATCH] Fix sa nas client. --- paddleslim/nas/sa_nas.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/paddleslim/nas/sa_nas.py b/paddleslim/nas/sa_nas.py index 3081b53f..00decbfd 100644 --- a/paddleslim/nas/sa_nas.py +++ b/paddleslim/nas/sa_nas.py @@ -66,10 +66,11 @@ class SANAS(object): if server_ip == None or server_ip == "": server_ip = self._get_host_ip() + factory = SearchSpaceFactory() + self._search_space = factory.get_search_space(configs) + # create controller server if self._is_server: - factory = SearchSpaceFactory() - self._search_space = factory.get_search_space(configs) init_tokens = self._search_space.init_tokens() range_table = self._search_space.range_table() range_table = (len(range_table) * [0], range_table) -- GitLab