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

Provide the option to return partial batches #123

Open
maxrjones opened this issue Nov 4, 2022 · 0 comments · May be fixed by #177
Open

Provide the option to return partial batches #123

maxrjones opened this issue Nov 4, 2022 · 0 comments · May be fixed by #177
Labels

Comments

@maxrjones
Copy link
Member

maxrjones commented Nov 4, 2022

Is your feature request related to a problem?

As explained in #82, xbatcher throws away partial batches if the input_dims dimension is not evenly divisible by the value provided.

Describe the solution you'd like

As noted in

# TODO: decide how to handle bsizes like 15 that don't evenly divide the dimension
# Should we enforce that each batch size always has to be the same
we could consider adding a parameter that specifies that the batch generator should also return partial batches.

Describe alternatives you've considered

If there is no strong need for this feature, we could keep the status-quo of only returning equal-sized batches.

Additional context

No response

@arbennett arbennett linked a pull request Mar 21, 2023 that will close this issue
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant