fix : _save_log 前确保日志目录存在，防止跨进程反序列化后目录丢失

2026-06-12 15:39:54 +08:00 · 2026-06-12 15:39:54 +08:00 · daf627a6de
parent 445378667f
commit daf627a6de
1 changed files with 1 additions and 0 deletions
--- a/astrai/trainer/train_callback.py
+++ b/astrai/trainer/train_callback.py
@ -271,6 +271,7 @@ class MetricLoggerCallback(TrainCallback):
    @only_on_rank(0)
    def _save_log(self, epoch, iter):
        log_file = self.log_dir / f"epoch_{epoch}_iter_{iter}_metric.jsonl"
+        log_file.parent.mkdir(parents=True, exist_ok=True)

        with open(log_file, "w") as f:
            for log in self.log_cache: