From d6bfb94215ac8a74f8c16a62a635250335719160 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 26 Jul 2019 18:52:38 -0400 Subject: [PATCH] added global rank var name --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index b2881db7..44cafb0c 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -501,7 +501,7 @@ class Trainer(TrainerIO): self.world_size = self.nb_gpu_nodes * len(self.data_parallel_device_ids) # let the exp know the rank to avoid overwriting logs - self.experiment.rank = self.global_rank + self.experiment.rank = self.proc_rank # set up server using proc 0's ip address # try to init for 20 times at max in case ports are taken