added global rank var name

2019-07-26 18:52:38 -04:00 · 2019-07-26 18:52:38 -04:00 · d6bfb94215
parent 12f717ad4a
commit d6bfb94215
1 changed files with 1 additions and 1 deletions
--- 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