From 64d41a0631cf27a9d66a861b1ce50329ca736fe4 Mon Sep 17 00:00:00 2001 From: wanghaoshuang Date: Tue, 26 Nov 2019 15:31:52 +0800 Subject: [PATCH] Fix sa controller. --- paddleslim/common/controller_server.py | 2 ++ paddleslim/common/sa_controller.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/paddleslim/common/controller_server.py b/paddleslim/common/controller_server.py index bf3ee3ab..eb37fe91 100644 --- a/paddleslim/common/controller_server.py +++ b/paddleslim/common/controller_server.py @@ -107,6 +107,8 @@ class ControllerServer(object): _logger.debug("send message to {}: [{}]".format(addr, tokens)) conn.close() + except Exception, err: + _logger.error(err) finally: self._socket_server.close() self.close() diff --git a/paddleslim/common/sa_controller.py b/paddleslim/common/sa_controller.py index 932f5b5d..f3dfa4b9 100644 --- a/paddleslim/common/sa_controller.py +++ b/paddleslim/common/sa_controller.py @@ -75,7 +75,7 @@ class SAController(EvolutionaryController): iter = int(iter) if iter > self._iter: self._iter = iter - temperature = self._init_temperature * self._reduce_rate**self._iter + temperature = self._init_temperature * self._reduce_rate**self._iter if (reward > self._reward) or (np.random.random() <= math.exp( (reward - self._reward) / temperature)): self._reward = reward -- GitLab