From 7070f3969da2dcaecfbaa8bf008aca9dfdbbb545 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sun, 15 Oct 2023 05:02:48 +0800 Subject: [PATCH] tiny fix Former-commit-id: 47b7b34357708a5354d542ddc239146c6417d718 --- src/llmtuner/extras/callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/extras/callbacks.py b/src/llmtuner/extras/callbacks.py index b2987605..7f7ce08e 100644 --- a/src/llmtuner/extras/callbacks.py +++ b/src/llmtuner/extras/callbacks.py @@ -135,7 +135,7 @@ class LogCallback(TrainerCallback): remaining_time=self.remaining_time ) if self.runner is not None: - logger.info("{{'loss': {:.4f}, 'learning_rate': {:2.2e}, 'epoch': {:.2f}}}".format( + logger.info("{{'loss': {:.4f}, 'learning_rate': {:2.4e}, 'epoch': {:.2f}}}".format( logs["loss"], logs["learning_rate"], logs["epoch"] ))