[stephantul]

The scheduler accidentally only stepped once per epoch. Because almost all our training tests are run on relatively small datasets (i.e., 10k-100k items), this was missed, and looked like it was working. The new scheduler steps every validation step

[greptile-apps]

Confidence Score: 4/5

[Medium impact] The change affects model training across the shared training workflows.

The PR appears safe to merge, with a non-blocking update needed to the training guide.

_{Reviews (1) · Last reviewed commit: "fix: scheduler should step more often" · Reviewed by Greptile}

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.

Files with missing lines	Coverage Δ
model2vec/train/trainer.py	99.08% <100.00%> (-0.05%)	⬇️

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.