Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

structuredconfig for train.py and eval.py #1051

Merged
merged 240 commits into from
May 8, 2024

Conversation

milocress
Copy link
Contributor

@milocress milocress commented Mar 22, 2024

Replace pop statements with omegaconf.structuredconfig initialization

  • Regression tests: llm-foundry-regression-tests-runner-mtlkuR llm-foundry-regression-tests-runner-FXx9Gb llm-foundry-regression-tests-runner-ZShXQp

Unit tests with variables:

  • tests/data/test_dataloader.py::test_correct_padding
  • tests/a_scripts/eval/test_eval.py::test_loader_eval
  • test/a_scripts/train/test_train.py::test_train_gauntlet

Manual tests with variables (latest): mpt-125m-chinchilla-regression-bN2HOr

mpt-125m-chinchilla-J1XkBi

mpt-125m-chinchilla-regression-kSFIMk

@milocress milocress marked this pull request as ready for review April 8, 2024 16:52
@milocress milocress requested a review from irenedea April 9, 2024 18:19
@milocress milocress requested a review from dakinggg May 7, 2024 20:55
Copy link
Collaborator

@dakinggg dakinggg left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, but please coordinate with @ShashankMosaicML to minimize the difficulty of his PR

@milocress milocress enabled auto-merge (squash) May 8, 2024 19:48
@mvpatel2000 mvpatel2000 dismissed irenedea’s stale review May 8, 2024 20:18

Stale, removing hold

@milocress milocress merged commit cc8351c into mosaicml:main May 8, 2024
9 checks passed
@milocress milocress deleted the milo/foundry-type-cleanup branch May 8, 2024 20:18
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.

None yet

3 participants