jrbourbeau / dask-optuna

Scale Optuna with Dask
https://jrbourbeau.github.io/dask-optuna/
MIT License
35 stars 6 forks source link

Remove batch_size from optimize #9

Closed jrbourbeau closed 3 years ago

jrbourbeau commented 3 years ago

Submitting trials in batches instead of individually doesn't improve performance noticeably for ~a few thousand trials. As I'd prefer to keep things simple for now, this PR removes the batch_size option from dask_optuna.optimize and instead submits trials individually. It'll be straightforward to add this back if there are users that end up wanting to use trial batches