set local and global rank

Lightning-AI · May 6, 2020 · 743bc4a · 743bc4a
1 parent c0a4f9d
commit 743bc4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -443,8 +443,8 @@ def __init__(
 
         # tpu state flags
         self.use_tpu = False
-        self.tpu_local_core_rank = None
-        self.tpu_global_core_rank = None
+        self.tpu_local_core_rank = self.tpu_id
+        self.tpu_global_core_rank = self.tpu_id
 
         # distributed backend choice
         self.distributed_backend = distributed_backend