fix eval resuming in webui

Former-commit-id: b28b53cd06777f213ef7b925a914ff5fd357ade1
This commit is contained in:
hiyouga
2023-10-15 15:45:38 +08:00
parent 7070f3969d
commit 68330eab2a
5 changed files with 27 additions and 22 deletions

View File

@@ -136,7 +136,7 @@ class LogCallback(TrainerCallback):
)
if self.runner is not None:
logger.info("{{'loss': {:.4f}, 'learning_rate': {:2.4e}, 'epoch': {:.2f}}}".format(
logs["loss"], logs["learning_rate"], logs["epoch"]
logs["loss"] or 0, logs["learning_rate"] or 0, logs["epoch"] or 0
))
os.makedirs(args.output_dir, exist_ok=True)