From 04e68f022fc03dd5f1555ee86dea997d42a448ad Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 30 Jun 2020 17:20:35 -0400 Subject: [PATCH] fix tpu tests --- pytorch_lightning/trainer/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index c97dbcfa2808f..fdb61cd9b7d72 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -1193,7 +1193,7 @@ def test( # attempt to load weights from a spawn path = os.path.join(self.default_root_dir, '__temp_weight_ddp_end.ckpt') test_model = self.model - if os.path.exists(path): + if os.path.exists(path) and self.on_colab_kaggle: test_model = self.load_spawn_weights(self.model) self.fit(test_model)