wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

testing master_Addr flag

Browse Source
This commit is contained in:
William Falcon
2019-07-12 11:43:05 -04:00
parent c223960edb
commit 6dde1d7ae3
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+1 -2
View File
@@ -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