Apply suggestions from code review

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Lightning-AI · Jul 27, 2020 · 3e70063 · 3e70063
1 parent 4032446
commit 3e70063
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: