From 784a05379355b9f8e341e2c0493ca71e2254d1d2 Mon Sep 17 00:00:00 2001 From: Vadim Bereznyuk Date: Sat, 1 Feb 2020 21:36:50 +0300 Subject: [PATCH] Fix default ckpt path when logger exists (#771) --- pytorch_lightning/trainer/callback_config.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/callback_config.py b/pytorch_lightning/trainer/callback_config.py index 7d0496c88b49f..3756b19e433c0 100644 --- 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"