Open
Description
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