Update pytorch_lightning/trainer/trainer.py

four4fish · tchaton · web-flow · commit f02be7c9b610 · 2021-12-09T10:13:45.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
@@ -1668,7 +1668,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
`@@ -1668,7 +1668,7 @@ def _log_device_info(self) -> None:`
`1668`	`1668`	`category=PossibleUserWarning,`
`1669`	`1669`	`)`
`1670`	`1670`
`1671`		`- if _TPU_AVAILABLE and isinstance(self.accelerator, TPUAccelerator):`
	`1671`	`+ if _TPU_AVAILABLE and not isinstance(self.accelerator, TPUAccelerator):`
`1672`	`1672`	`rank_zero_warn(`
`1673`	`1673`	"TPU available but not used. Set the `tpu_cores` flag in your trainer"
`1674`	`1674`	" `Trainer(tpu_cores=8)` or script `--tpu_cores=8`."