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

cleaning up demos

Browse Source
This commit is contained in:
William Falcon
2019-10-05 16:19:53 -04:00
parent d179ad6d1d
commit 0d86302e05
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/trainer.py
-4
View File
@@ -828,8 +828,6 @@ class Trainer(TrainerIO):
except Exception:
self.node_rank = 0
print('node rank:', self.node_rank)
# show progressbar only on progress_rank 0
self.show_progress_bar = self.show_progress_bar and self.node_rank == 0 and gpu_nb == 0
@@ -933,8 +931,6 @@ class Trainer(TrainerIO):
root_node = self.resolve_root_node_address(root_node)
os.environ['MASTER_ADDR'] = root_node
print(default_port, self.proc_rank, self.world_size)
dist.init_process_group("nccl", rank=self.proc_rank, world_size=self.world_size)
def resolve_root_node_address(self, root_node):