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

removed logging

Browse Source
This commit is contained in:
William Falcon committed 2019-07-20 09:22:47 -04:00
1 parent 1a39f703ad
commit 0fdf290201
1 file changed
-4
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
-4
View File
@@ -512,10 +512,6 @@ class Trainer(TrainerIO):
root_node = self.__resolve_root_node_address()
os.environ['MASTER_ADDR'] = root_node
print('-'*100)
print(f'INIT RANK: {self.proc_rank}, NODE:{self.node_rank}, WORLD_SIZE:{self.world_size}, ADDR: {root_node}, PORT: {port}')
print('-'*100)
dist.init_process_group("nccl", rank=self.proc_rank, world_size=self.world_size)
def __resolve_root_node_address(self):