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

Eval subset num batches bug fix #2028

Merged
merged 1 commit into from
Mar 3, 2023

Conversation

mvpatel2000
Copy link
Contributor

@mvpatel2000 mvpatel2000 commented Mar 3, 2023

What does this PR do?

Currently, trainer ignores eval_subset_num_batches when passed into init when calling trainer.eval() and always uses subset_num_batches. Instead, we should use subset_num_batches if specified, or otherwise fall back on what is already specified in the evaluator (either through trainer or in evaluator constructor). This PR fixes this and adds a unit test for it.

What issue(s) does this change relate to?

CO-1753

@mvpatel2000 mvpatel2000 requested review from dakinggg and eracah and removed request for dakinggg March 3, 2023 21:35
Copy link
Contributor

@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

@mvpatel2000 mvpatel2000 merged commit 11d1c51 into mosaicml:dev Mar 3, 2023
@mvpatel2000 mvpatel2000 deleted the mvpatel2000/eval-subset branch March 3, 2023 22:31
bandish-shah pushed a commit that referenced this pull request Mar 5, 2023
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