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)
