removed form trainer and early stopping

Lightning-AI · Sep 17, 2020 · ef11533 · ef11533
1 parent 9907ad3
commit ef11533
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 7 deletions.
diff --git a/pytorch_lightning/callbacks/early_stopping.py b/pytorch_lightning/callbacks/early_stopping.py
@@ -34,9 +34,6 @@
 
 torch_inf = torch.tensor(np.Inf)
 
-if TPU_AVAILABLE:
-    import torch_xla.core.xla_model as xm
-    import torch_xla
 
 
 

diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -59,10 +59,6 @@
 from pytorch_lightning.trainer.properties import TrainerProperties
 
 # warnings to ignore in trainer
-from pytorch_lightning.utilities.xla_device_utils import tpu_device_exists
-
-TPU_AVAILABLE = tpu_device_exists()
-
 warnings.filterwarnings(
     'ignore', message='torch.distributed.reduce_op is deprecated, ' 'please use torch.distributed.ReduceOp instead'
 )