Skip to content

Commit 784a053

Browse files
authored
Fix default ckpt path when logger exists (#771)
1 parent 9a6838d commit 784a053

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pytorch_lightning/trainer/callback_config.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,11 @@ def configure_checkpoint_callback(self):
2323
if self.checkpoint_callback is True:
2424
# init a default one
2525
if self.logger is not None:
26+
save_dir = (getattr(self.logger, 'save_dir', None) or
27+
getattr(self.logger, '_save_dir', None) or
28+
self.default_save_path)
2629
ckpt_path = os.path.join(
27-
self.default_save_path,
30+
save_dir,
2831
self.logger.name,
2932
f'version_{self.logger.version}',
3033
"checkpoints"

0 commit comments

Comments
 (0)