From d66afebf8b8bcb26d1e2ef3c9af2da9b8610df3e Mon Sep 17 00:00:00 2001
From: zhouzj <41366441+zzjjay@users.noreply.github.com>
Date: Mon, 21 Mar 2022 14:20:38 +0800
Subject: [PATCH] fix nas's bug (#1019)

---
 paddleslim/common/controller_client.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/paddleslim/common/controller_client.py b/paddleslim/common/controller_client.py
index ceb6ecf54ac82..526f348b2563a 100644
--- a/paddleslim/common/controller_client.py
+++ b/paddleslim/common/controller_client.py
@@ -17,7 +17,7 @@
 import logging
 import socket
 from .log_helper import get_logger
-
+import sys
 __all__ = ['ControllerClient']
 
 _logger = get_logger(__name__, level=logging.INFO)
@@ -110,7 +110,10 @@ def next_tokens(self):
 
         socket_client.send("next_tokens".encode())
         tokens = socket_client.recv(1024).decode()
-        tokens = [int(token) for token in tokens.strip("\n").split(",")]
+        try:
+            tokens = [int(token) for token in tokens.strip("\n").split(",")]
+        except ValueError:
+            sys.exit()
         return tokens
 
     def request_current_info(self):