testing master_Addr flag

2019-07-12 11:30:57 -04:00 · 2019-07-12 11:30:57 -04:00 · c223960edb
parent 32646cf2ee
commit c223960edb
1 changed files with 2 additions and 1 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -336,7 +336,8 @@ 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 = self.__get_root_node_ip(self.proc_rank, self.nb_gpu_nodes)
+        ip = os.environ['MASTER_ADDR']
        self.__init_tcp_connection(ip)

        # CHOOSE OPTIMIZER