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

Upload wandb results #164

Merged
merged 1 commit into from
Jun 13, 2024
Merged

Upload wandb results #164

merged 1 commit into from
Jun 13, 2024

Conversation

wongjingping
Copy link
Collaborator

@wongjingping wongjingping commented Jun 13, 2024

Added notebook for uploading vllm eval results to wandb post-hoc.
This notebook will take the results produced by run_checkpoints.sh and run_checkpoints_cot.sh and upload them to the specified run id on wandb.
There is 1 weird quirk about wandb where they don't allow you to log to past steps once completed (see forum post). To work around that, we just continue logging from the latest step onwards (which will cause our steps displayed on wandb to be inconsistent with the actual checkpoints, but should be sufficient for comparison reasons).
For example, you can see that the eval for each step starts after 1k steps in the screenshot below:

wandb vllm

@rishsriv
Copy link
Member

YESSSS! Thank you!

@wongjingping wongjingping merged commit 6690087 into main Jun 13, 2024
2 checks passed
@wongjingping wongjingping deleted the jp/upload_wandb branch June 13, 2024 04:48
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.

2 participants