Skip to content

Update evo2 ModelCheckpoint args#935

Merged
jwilber merged 5 commits into
mainfrom
jwilber/evo2-add-save-top-k-arg
Jun 10, 2025
Merged

Update evo2 ModelCheckpoint args#935
jwilber merged 5 commits into
mainfrom
jwilber/evo2-add-save-top-k-arg

Conversation

@jwilber
Copy link
Copy Markdown
Collaborator

@jwilber jwilber commented Jun 9, 2025

Description

Adds new arguments to control ModelCheckpoint.

Type of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Refactor
  • Documentation update
  • Other (please describe):

Pre-submit Checklist

  • I have tested these changes locally
  • I have updated the documentation accordingly
  • I have added/updated tests as needed
  • All existing tests pass successfully

Signed-off-by: Jared Wilber <jwilber@nvidia.com>
@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Jun 9, 2025

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

Signed-off-by: Jared Wilber <jwilber@nvidia.com>
@jwilber jwilber force-pushed the jwilber/evo2-add-save-top-k-arg branch from 74eb26f to 96b6bc4 Compare June 9, 2025 19:18
@jwilber
Copy link
Copy Markdown
Collaborator Author

jwilber commented Jun 9, 2025

/ok to test 96b6bc4

@codecov-commenter
Copy link
Copy Markdown

codecov-commenter commented Jun 9, 2025

Codecov Report

Attention: Patch coverage is 0% with 5 lines in your changes missing coverage. Please review.

Project coverage is 84.17%. Comparing base (5aa0d3b) to head (9b64b79).
Report is 1 commits behind head on main.

✅ All tests successful. No failed tests found.

Files with missing lines Patch % Lines
...ackages/bionemo-evo2/src/bionemo/evo2/run/train.py 0.00% 5 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main     #935      +/-   ##
==========================================
- Coverage   84.21%   84.17%   -0.04%     
==========================================
  Files         143      143              
  Lines        9061     9064       +3     
==========================================
- Hits         7631     7630       -1     
- Misses       1430     1434       +4     
Files with missing lines Coverage Δ
...ackages/bionemo-evo2/src/bionemo/evo2/run/train.py 15.76% <0.00%> (-0.82%) ⬇️

... and 1 file with indirect coverage changes

jwilber added 3 commits June 10, 2025 11:26
Signed-off-by: Jared Wilber <jwilber@nvidia.com>
Signed-off-by: Jared Wilber <jwilber@nvidia.com>
@jwilber
Copy link
Copy Markdown
Collaborator Author

jwilber commented Jun 10, 2025

/ok to test 9b64b79

@jwilber jwilber enabled auto-merge June 10, 2025 18:59
@jwilber jwilber changed the title Add arg for save_top_k to nemo checkpoint Update evo2 ModelCheckpoint args Jun 10, 2025
@jwilber jwilber added this pull request to the merge queue Jun 10, 2025
Merged via the queue into main with commit 8d3d043 Jun 10, 2025
10 checks passed
@jwilber jwilber deleted the jwilber/evo2-add-save-top-k-arg branch June 10, 2025 21:52
camirr-nv pushed a commit that referenced this pull request Jun 26, 2025
### Description
Adds new arguments to control `ModelCheckpoint`.

### Type of changes
<!-- Mark the relevant option with an [x] -->

- [ ]  Bug fix (non-breaking change which fixes an issue)
- [x]  New feature (non-breaking change which adds functionality)
- [ ]  Refactor
- [x]  Documentation update
- [ ]  Other (please describe):

### Pre-submit Checklist
<!--- Ensure all items are completed before submitting -->

 - [x] I have tested these changes locally
 - [x] I have updated the documentation accordingly
 - [x] I have added/updated tests as needed
 - [x] All existing tests pass successfully

---------

Signed-off-by: Jared Wilber <jwilber@nvidia.com>
Signed-off-by: Ubuntu <camirr@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants