From a00e24ed805b5658eb0b1c7e31fad08ae1288b9d Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 5 Oct 2019 16:09:48 -0400 Subject: [PATCH] cleaning up demos --- pytorch_lightning/trainer/trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index ee6c24af..54dffd99 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -919,7 +919,7 @@ class Trainer(TrainerIO): # if user gave a port number, use that one instead try: - port = os.environ['MASTER_PORT'] + default_port = os.environ['MASTER_PORT'] except Exception: os.environ['MASTER_PORT'] = str(default_port) @@ -931,6 +931,7 @@ class Trainer(TrainerIO): root_node = self.resolve_root_node_address(root_node) os.environ['MASTER_ADDR'] = root_node + print(default_port) dist.init_process_group("nccl", rank=self.proc_rank, world_size=self.world_size)