fix(tests): correct MLX smoke test losses_per_step assertion
logging_steps=1 with max_steps=30 produces 30 loss entries, not 7. The assertion was stale from a previous config.
This commit is contained in:
parent
ec021a03d1
commit
90f6cd417d
1 changed files with 1 additions and 1 deletions
|
|
@ -390,7 +390,7 @@ def cmd_train(args) -> int:
|
|||
)
|
||||
if k in train_result
|
||||
}
|
||||
assert len(losses_per_step) == 7, f"expected 7 logged steps, got {losses_per_step}"
|
||||
assert len(losses_per_step) == 30, f"expected 30 logged steps (logging_steps=1, max_steps=30), got {losses_per_step}"
|
||||
for i, l in enumerate(losses_per_step):
|
||||
# Allow exact 0.0: fp16 per-step loss underflows to 0.0 after
|
||||
# the LoRA reaches loss=0 around step ~10 with this fixture +
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue