(self, gpus, server)
| 228 | devices = None |
| 229 | |
| 230 | def __init__(self, gpus, server): |
| 231 | super(DistributedTrainerBase, self).__init__() |
| 232 | self.devices = gpus |
| 233 | self.server = server |
| 234 | self.job_name = server.server_def.job_name |
| 235 | logger.info("Distributed training on cluster:\n" + str(server.server_def.cluster)) |
| 236 | |
| 237 | def join(self): |
| 238 | logger.info("Calling server.join() on {}:{}".format(self.job_name, self.server.server_def.task_index)) |