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

Tuner.scale_batch_size(max_val=1024) #20364

Open
edmcman opened this issue Oct 24, 2024 · 0 comments
Open

Tuner.scale_batch_size(max_val=1024) #20364

edmcman opened this issue Oct 24, 2024 · 0 comments
Labels
feature Is an improvement or enhancement needs triage Waiting to be triaged by maintainers

Comments

@edmcman
Copy link

edmcman commented Oct 24, 2024

Description & Motivation

I'm always frustrated when scale_batch_size runs on a CPU machine, and ends up testing horribly large batch sizes like 2^25.

Pitch

I'd like to be able to pass an upper bound batch size as an argument. Batch sizes larger than the argument will never be tested.

Alternatives

No response

Additional context

No response

cc @Borda

@edmcman edmcman added feature Is an improvement or enhancement needs triage Waiting to be triaged by maintainers labels Oct 24, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature Is an improvement or enhancement needs triage Waiting to be triaged by maintainers
Projects
None yet
Development

No branches or pull requests

1 participant