Apply suggestions from code review

Co-authored-by: Rohit Gupta <[email protected]>
Lightning-AI · Jul 27, 2020 · 6c9495e · 6c9495e
1 parent 45233a5
commit 6c9495e
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: