zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 12:49:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

LoRA: report last train info (#595)

Browse Source
This commit is contained in:
madroid
2024-03-20 08:29:50 +08:00
committed by GitHub
parent 4680ef4413
commit 39d5ca6427
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -224,7 +224,7 @@ def train(
n_tokens += toks.item()
# Report training loss if needed
if (it + 1) % args.steps_per_report == 0:
if ((it + 1) % args.steps_per_report == 0) or (it + 1 == args.iters):
train_loss = np.mean(losses)
stop = time.perf_counter()
@@ -259,7 +259,7 @@ def train(
start = time.perf_counter()
# Report validation loss if needed
if it == 0 or (it + 1) % args.steps_per_eval == 0:
if it == 0 or ((it + 1) % args.steps_per_eval == 0) or (it + 1 == args.iters):
stop = time.perf_counter()
val_loss = evaluate(
model=model,