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

bugfix: remove out large samples from the multi pack batch sampler #52

Merged
merged 1 commit into from
Jun 24, 2024

Conversation

ChanderG
Copy link
Contributor

Fix for #51

larger entries, longer than max_batch_len, may be present in the output of the sampler which gets filtered out in the collator

in rare instances, this can lead to empty batches which can be problematic for the training loop

@aldopareja
Copy link
Member

Great catch! This should be processed in the data processing step. Can you add a warning in red or yellow 033 notation so the user knows?

@aldopareja
Copy link
Member

You can just add it when there's a sample larger than Max batch lenght

larger entries, longer than max_batch_len, may be present in the
output of the sampler which gets filtered out in the collator

in rare instances, this can lead to empty batches which can be
problematic for the training loop

Signed-off-by: ChanderG <[email protected]>
@aldopareja aldopareja merged commit ee76c17 into instructlab:main Jun 24, 2024
5 of 6 checks passed
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