Fix default ckpt path when logger exists (#771)

2026-09-09 11:32:07 +08:00 · 2020-02-01 13:36:50 -05:00
parent 9a6838d349
commit 784a053793
1 changed files with 4 additions and 1 deletions
@@ -23,8 +23,11 @@ class TrainerCallbackConfigMixin(ABC):
        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"