"archive-del/deployment/vscode:/vscode.git/clone" did not exist on "0f0c6288c35a456707057b0aab7754fcb5ac4ffd"
Unverified Commit 7d92009a authored by Deep Gandhi's avatar Deep Gandhi Committed by GitHub
Browse files

Added additional kwarg for successful running of optuna hyperparameter search (#31924)

Update integration_utils.py

Added additional kwarg
parent 63700628
......@@ -252,10 +252,11 @@ def run_hp_search_optuna(trainer, n_trials: int, direction: str, **kwargs) -> Be
timeout = kwargs.pop("timeout", None)
n_jobs = kwargs.pop("n_jobs", 1)
gc_after_trial = kwargs.pop("gc_after_trial", False)
directions = direction if isinstance(direction, list) else None
direction = None if directions is not None else direction
study = optuna.create_study(direction=direction, directions=directions, **kwargs)
study.optimize(_objective, n_trials=n_trials, timeout=timeout, n_jobs=n_jobs)
study.optimize(_objective, n_trials=n_trials, timeout=timeout, n_jobs=n_jobs, gc_after_trial=gc_after_trial)
if not study._is_multi_objective():
best_trial = study.best_trial
return BestRun(str(best_trial.number), best_trial.value, best_trial.params)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment