From a21dc5a187210d85bb89a3e7b577317496808515 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 11 Jul 2019 15:15:22 -0400 Subject: [PATCH] simplify trainer output --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 39738e0a9b..fb99dda22a 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -360,7 +360,7 @@ class Trainer(TrainerIO): raise RuntimeError('Failed to connect using 20 different ip addresses') try: - dist.init_process_group("nccl", init_method=f'tcp://{ip}:12001', rank=self.proc_rank, world_size=self.world_size) + dist.init_process_group("nccl", init_method=f'tcp://{ip}:{port}', rank=self.proc_rank, world_size=self.world_size) except RuntimeError as e: # port taken warnings.warn(f'port {port} taken, trying port {port}...')