def action_after(self, worker_id, message):
# override PTServer class method, for training related action
PTServer.action_after(self, worker_id, message )
if message in ['connect', 'sync_register']:
self.prepare_param_exchanger()
elif message == 'exchange':
self.exchanger.comm = self.worker_comm[str(worker_id)]
self.exchanger.dest = self.worker_rank[str(worker_id)]
self.exchanger.exchange()
elif message == 'copy_to_local':
self.exchanger.comm = self.worker_comm[str(worker_id)]
self.exchanger.dest = self.worker_rank[str(worker_id)]
self.exchanger.copy_to_local()
def action_after(self, worker_id, message):
# override PTServer class method, for training related action
PTServer.action_after(self, worker_id, message)
if message in ['connect', 'sync_register']:
self.prepare_param_exchanger()
elif message == 'exchange':
self.exchanger.comm = self.worker_comm[str(worker_id)]
self.exchanger.dest = self.worker_rank[str(worker_id)]
self.exchanger.exchange()
elif message == 'copy_to_local':
self.exchanger.comm = self.worker_comm[str(worker_id)]
self.exchanger.dest = self.worker_rank[str(worker_id)]
self.exchanger.copy_to_local()
