From 1225fd50160dac1a3070e12ba424391eccbd788e Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Fri, 22 Nov 2019 15:11:19 +0800
Subject: [PATCH] Add tokens2arch function in sa nas class.

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

diff --git a/paddleslim/nas/sa_nas.py b/paddleslim/nas/sa_nas.py
index b9dca29b..3081b53f 100644
--- a/paddleslim/nas/sa_nas.py
+++ b/paddleslim/nas/sa_nas.py
@@ -100,6 +100,9 @@ class SANAS(object):
     def _get_host_ip(self):
         return socket.gethostbyname(socket.gethostname())
 
+    def tokens2arch(self, tokens):
+        return self._search_space.token2arch(self.tokens)
+
     def next_archs(self):
         """
         Get next network architectures.
-- 
GitLab