skip best_model_path if checkpoint_callback is None (#2962)

* skip best_model_path if checkpoint_callback is None * removed test
Lightning-AI · Oct 1, 2020 · 8be002c · 8be002c
1 parent e4e60e9
commit 8be002c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytorch_lightning/accelerators/tpu_backend.py b/pytorch_lightning/accelerators/tpu_backend.py
@@ -67,7 +67,8 @@ def teardown(self):
         last_path = self.mp_queue.get()
 
         # transfer back the best path to the trainer
-        self.trainer.checkpoint_callback.best_model_path = best_path
+        if self.trainer.checkpoint_callback is not None:
+            self.trainer.checkpoint_callback.best_model_path = best_path
         # todo, pass also bets score
 
         # load last weights