diff --git a/paddleslim/nas/sa_nas.py b/paddleslim/nas/sa_nas.py
index b9dca29b49f5f298a740ca5d40e99d27c5558de7..3081b53f818bbf3c179b0f4f4dbc60a8e7b7089b 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.