Fix default ckpt path when logger exists (#771)

Lightning-AI · Feb 1, 2020 · 784a053 · 784a053
1 parent 9a6838d
commit 784a053
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pytorch_lightning/trainer/callback_config.py b/pytorch_lightning/trainer/callback_config.py
@@ -23,8 +23,11 @@ def configure_checkpoint_callback(self):
         if self.checkpoint_callback is True:
             # init a default one
             if self.logger is not None:
+                save_dir = (getattr(self.logger, 'save_dir', None) or
+                            getattr(self.logger, '_save_dir', None) or
+                            self.default_save_path)
                 ckpt_path = os.path.join(
-                    self.default_save_path,
+                    save_dir,
                     self.logger.name,
                     f'version_{self.logger.version}',
                     "checkpoints"