Skip to content

model.autotune

Optuna-based autotuning helpers used by train.py.

Optuna-based hyperparameter autotuning helpers for training.

Classes

Functions

run_autotune(args, device, *, trainer_cls, data_cache_path_fn, load_data_cache_fn, save_data_cache_fn, check_watchlist_coverage_fn, watchlist_species)

Run Optuna hyperparameter search and print best parameters.