Update pytorch_lightning/trainer/trainer.py

four4fish · tchaton · four4fish · commit a5117e01ed81 · 2021-12-09T11:48:49.000-08:00
Co-authored-by: thomas chaton &lt;thomas@grid.ai&gt;
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -1644,7 +1644,7 @@ def _log_device_info(self) -> None:
                 category=PossibleUserWarning,
             )
 
-        if _TPU_AVAILABLE and isinstance(self.accelerator, TPUAccelerator):
+        if _TPU_AVAILABLE and not isinstance(self.accelerator, TPUAccelerator):
             rank_zero_warn(
                 "TPU available but not used. Set the `tpu_cores` flag in your trainer"
                 " `Trainer(tpu_cores=8)` or script `--tpu_cores=8`."

Original file line number	Diff line number	Diff line change
`@@ -1644,7 +1644,7 @@ def _log_device_info(self) -> None:`
`1644`	`1644`	`category=PossibleUserWarning,`
`1645`	`1645`	`)`
`1646`	`1646`
`1647`		`- if _TPU_AVAILABLE and isinstance(self.accelerator, TPUAccelerator):`
	`1647`	`+ if _TPU_AVAILABLE and not isinstance(self.accelerator, TPUAccelerator):`
`1648`	`1648`	`rank_zero_warn(`
`1649`	`1649`	"TPU available but not used. Set the `tpu_cores` flag in your trainer"
`1650`	`1650`	" `Trainer(tpu_cores=8)` or script `--tpu_cores=8`."