From e6f311bd8be8d7ba0fef5c8f482351374ab2ae1f Mon Sep 17 00:00:00 2001 From: "jones.wz" Date: Mon, 8 Aug 2022 10:58:33 +0800 Subject: [PATCH] fixed client state error --- federatedscope/autotune/fedex/server.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/federatedscope/autotune/fedex/server.py b/federatedscope/autotune/fedex/server.py index 3094f1f04..cfbcad6fd 100644 --- a/federatedscope/autotune/fedex/server.py +++ b/federatedscope/autotune/fedex/server.py @@ -158,6 +158,8 @@ def broadcast_model_para(self, else: # broadcast to all clients receiver = list(self.comm_manager.neighbors.keys()) + if msg_type == 'model_para': + self.sampler.change_state(receiver, 'working') if self._noise_injector is not None and msg_type == 'model_para': # Inject noise only when broadcast parameters @@ -198,6 +200,7 @@ def broadcast_model_para(self, def callback_funcs_model_para(self, message: Message): round, sender, content = message.state, message.sender, message.content + self.sampler.change_state(sender, 'idle') # For a new round if round not in self.msg_buffer['train'].keys(): self.msg_buffer['train'][round] = dict()