fix eval resuming in webui

Former-commit-id: 273745f9b9
This commit is contained in:
hiyouga
2023-10-15 15:45:38 +08:00
parent 99592478c9
commit 0503d45782
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)