diff --git a/paddleslim/common/controller_client.py b/paddleslim/common/controller_client.py index fd6575e04b16d6491ad02dc7a6e1725c68505460..ad989dd16014fa8e6fa1495516e81048324fb826 100644 --- a/paddleslim/common/controller_client.py +++ b/paddleslim/common/controller_client.py @@ -51,7 +51,7 @@ class ControllerClient(object): socket_client.send("{}\t{}\t{}".format(self._key, tokens, reward) .encode()) response = socket_client.recv(1024).decode() - if response.trip('\n').split("\t") == "ok": + if response.strip('\n').split("\t") == "ok": return True else: return False