tiny fix

Chen-mingxuan · Oct 14, 2023 · 3ad8c92 · 3ad8c92
1 parent 1e94017
commit 3ad8c92
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/llmtuner/extras/callbacks.py b/src/llmtuner/extras/callbacks.py
@@ -135,7 +135,7 @@ def on_log(self, args: "TrainingArguments", state: "TrainerState", control: "Tra
             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"]
             ))