From 6c9495e04bbe602cb662fac8376c77dfe2280922 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Mon, 27 Jul 2020 23:05:36 +0200 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Rohit Gupta --- pytorch_lightning/accelerator_backends/tpu_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/accelerator_backends/tpu_backend.py b/pytorch_lightning/accelerator_backends/tpu_backend.py index 95203550eb37e..8d1d1b271b7dc 100644 --- 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: