Merge branch 'tpu/fix-tests' of https://github.com/PyTorchLightning/p…

…ytorch-lightning into tpu/fix-tests
Lightning-AI · Jul 27, 2020 · 498cf3a · 498cf3a
2 parents 7fc67cf + 3e70063
commit 498cf3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/accelerator_backends/tpu_backend.py b/pytorch_lightning/accelerator_backends/tpu_backend.py
@@ -126,7 +126,7 @@ def __save_end_of_training_weights(self, model: LightningModule, trainer):
 
     def __setup_tpu_training(self, model: LightningModule, trainer):
         # use the default device from the process
-        tpu_device = xm.xla_device()
+        # tpu_device = xm.xla_device()
 
         # if given an ordinal device, use this as the device
         if trainer.tpu_id is not None: