From 6dde1d7ae3d6286c6bb2c10168169aff248460ee Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 12 Jul 2019 11:43:05 -0400 Subject: [PATCH] testing master_Addr flag --- pytorch_lightning/models/trainer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 6d4f639c..3ede697c 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -336,8 +336,7 @@ class Trainer(TrainerIO): # set up server using proc 0's ip address # try to init for 20 times at max in case ports are taken - # ip = self.__get_root_node_ip(self.proc_rank, self.nb_gpu_nodes) - ip = os.environ['MASTER_ADDR'] + ip = self.__get_root_node_ip(self.proc_rank, self.nb_gpu_nodes) self.__init_tcp_connection(ip) # CHOOSE OPTIMIZER