def still_wait(self): ps_result = remote_method(ParameterServer.still_wait, self.param_server_rref) if ps_result == 'exit': print(f'From master: training process completed.') os._exit(0) else: return ps_result
def update_ps_weights(self, new_weights): return remote_method( ParameterServer.update_weights, self.param_server_rref, new_weights)
def fetch_worker_K_from_ps(self): return remote_method( ParameterServer.fetch_worker_K, self.param_server_rref)
def fetch_model_type_from_ps(self): return remote_method( ParameterServer.fetch_model_type, self.param_server_rref)
def fetch_gamma_from_ps(self): return remote_method( ParameterServer.fetch_gamma, self.param_server_rref)
def fetch_avg_grad_grom_ps(self): return remote_method( ParameterServer.fetch_workers_avg_grad, self.param_server_rref)
def fetch_weights_from_ps(self): return remote_method( ParameterServer.fetch_weights, self.param_server_rref)
def still_wait(self): return remote_method( ParameterServer.still_wait, self.param_server_rref)
def send_msg_to_ps(self, msg): return remote_method( ParameterServer.process_msg, self.param_server_rref, msg)
def send_worker_info(self, grad, other_info={}): return remote_method( ParameterServer.receive_worker_info, self.param_server_rref, grad, other_info)