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

Min and max batch size #23

Open
almilder opened this issue May 17, 2023 · 0 comments
Open

Min and max batch size #23

almilder opened this issue May 17, 2023 · 0 comments
Assignees

Comments

@almilder
Copy link
Collaborator

Currently the batch size can have a minimum value of 2 and a maximum value of 7. The minimum value of 2 is due to how the fits are postprocessed. When a batch size of 1 is used it changes the shape of the weights dictionary causing removing an index. This will have to be resolved in order to allow for batch size of 1. The maximum batch size of 7 is set by memory constraints in calculating the hessian. When a larger batch size is used the hessian call in postprocess.recalculate_with_chosen_weights returns an OOM error.

@almilder almilder assigned almilder and joglekara and unassigned almilder Dec 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

No branches or pull requests

2 participants