If desired, training can be stopped on a specific step without impacting the LR curve.

[jstjohn]

See new option --early-stop-on-step which may be used in JET tests to stop on a specific step without impacting the LR curve by changing --max-steps

[dorotat-nv]

Still, would be nice to add a unit tests with label slow

[codecov-commenter]

Codecov Report

❌ Patch coverage is 53.84615% with 6 lines in your changes missing coverage. Please review.
✅ Project coverage is 84.15%. Comparing base (588fb75) to head (785aaf4).
⚠️ Report is 515 commits behind head on main.

Files with missing lines	Patch %	Lines
...ackages/bionemo-evo2/src/bionemo/evo2/run/train.py	20.00%	4 Missing ⚠️
...o-testing/src/bionemo/testing/testing_callbacks.py	75.00%	2 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #739      +/-   ##
==========================================
- Coverage   85.93%   84.15%   -1.78%     
==========================================
  Files         120      128       +8     
  Lines        7310     8029     +719     
==========================================
+ Hits         6282     6757     +475     
- Misses       1028     1272     +244     

Files with missing lines	Coverage Δ
...o-testing/src/bionemo/testing/testing_callbacks.py	96.11% <75.00%> (-1.85%)	⬇️
...ackages/bionemo-evo2/src/bionemo/evo2/run/train.py	17.12% <20.00%> (ø)

... and 1 file with indirect coverage changes

🚀 New features to boost your workflow:

• 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.