This commit is contained in:
hiyouga committed 2024-09-01 21:15:44 +08:00
1 parent 64cb947c60
commit 55027282cd
2 files changed
+7 -1

No files matched your search

+1 -1
View File
@@ -318,7 +318,7 @@ class LogCallback(TrainerCallback):
if self.webui_mode and all(key in logs for key in ["loss", "learning_rate", "epoch"]):
logger.info(
"{{'loss': {:.4f}, 'learning_rate': {:2.4e}, 'epoch': {:.2f}, 'throughput': {}}}".format(
logs["loss"], logs["learning_rate"], logs["epoch"], logs.get("throughput")
logs["loss"], logs["learning_rate"], logs["epoch"], logs.get("throughput", "N/A")
)
)