ViperEkura
/
AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

This commit is contained in:
ViperEkura 2026-06-12 15:39:54 +08:00
parent 445378667f
commit daf627a6de
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
astrai/trainer/train_callback.py
View File

@ -271,6 +271,7 @@ class MetricLoggerCallback(TrainCallback):
@only_on_rank(0) @only_on_rank(0)
def _save_log(self, epoch, iter): def _save_log(self, epoch, iter):
log_file = self.log_dir / f"epoch_{epoch}_iter_{iter}_metric.jsonl" 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: with open(log_file, "w") as f:
for log in self.log_cache: for log in self.log_cache: