diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index bbfafd89..44cafb0c 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -500,6 +500,9 @@ class Trainer(TrainerIO): self.proc_rank = self.node_rank * len(self.data_parallel_device_ids) + gpu_nb 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.proc_rank + # set up server using proc 0's ip address # try to init for 20 times at max in case ports are taken # where to store ip_table diff --git a/setup.py b/setup.py index 5edadfa1..89ac6f29 100755 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ setup( install_requires=[ "torch>=1.1.0", "tqdm", - "test-tube>=0.6.7.1", + "test-tube>=0.6.7.4", ], packages=find_packages(), long_description=open("README.md", encoding="utf-8").read(),