-
Notifications
You must be signed in to change notification settings - Fork 526
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
Output eval logging batch #961
Output eval logging batch #961
Conversation
* relax atol and add retries to reduce flakiness in lion8b timing test
… into output_eval_logging
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great work Max! this was a many months effort to get a design acceptable to everyone and it involved some very complex torch metrics related engineering. Kudos!
Might need to wait for composer release ( |
…lm-foundry into output_eval_logging_batch
@maxisawesome now that Composer 0.21 is out, is this still blocked? |
yes @nik-mosaic I'm running some tests and then will merge |
* Skip flaky lion8b test (#598) * relax atol and add retries to reduce flakiness in lion8b timing test * add eval output logging * add back tasks * foo * add rlhf prompts * add rlhf prompts * add rlhf prompts * add rlhf prompts * add rlhf prompts * fix prompt * fix prompt * modify mcli * test * test * fix * fix merge * wip * merge * reset files, wip commit * rm small changes * reduce changes * reduce changes * . * wip * rm batch keys * revert init device * linting * add import * fix import * add eval_output_logging to registry * readd import * pyright + linting --------- Co-authored-by: dblalock <[email protected]> Co-authored-by: Jeremy Dohmann <[email protected]>
PR with adding output eval logging callback for eval.py. Works with recent changes to composer
Required in your yaml:
test run here: `test-batch-logging-izC0mH
wandb logged results here: https://wandb.ai/mosaic-ml/test-eval-logging/runs/km61skch/workspace