zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 18:26:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Flush the messages

Browse Source
This commit is contained in:
Angelos Katharopoulos 2024-09-12 15:04:21 -07:00
parent e0f18d15aa
commit b0a42d091e
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
llms/mlx_lm/tuner/trainer.py
View File

@ -254,7 +254,8 @@ def train(
print(
f"Iter {it}: "
f"Val loss {val_loss:.3f}, "
f"Val took {val_time:.3f}s"
f"Val took {val_time:.3f}s",
flush=True,
)
if training_callback is not None:
@ -292,7 +293,8 @@ def train(
f"It/sec {it_sec:.3f}, "
f"Tokens/sec {tokens_sec:.3f}, "
f"Trained Tokens {trained_tokens}, "
f"Peak mem {peak_mem:.3f} GB"
f"Peak mem {peak_mem:.3f} GB",
flush=True,
)
if training_callback is not None: