p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tiny fix

This commit is contained in:
hiyouga 2023-10-15 05:02:48 +08:00
parent 1e9401744c
commit 3ad8c92eca
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/extras/callbacks.py
View File

@ -135,7 +135,7 @@ class LogCallback(TrainerCallback):
remaining_time=self.remaining_time remaining_time=self.remaining_time
) )
if self.runner is not None: 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"] logs["loss"], logs["learning_rate"], logs["epoch"]
)) ))