riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

Fix formatting

Browse Source
This commit is contained in:
Miloš Mičík 2024-11-14 13:17:43 +00:00 committed by GitHub
parent 1f2938491f
commit 1217a9ff58
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/transformers/integrations/integration_utils.py
View File

@ -259,7 +259,14 @@ def run_hp_search_optuna(trainer, n_trials: int, direction: str, **kwargs) -> Be
direction = None if directions is not None else direction
optuna_callbacks = kwargs.pop("callbacks", None)
study = optuna.create_study(direction=direction, directions=directions, **kwargs)
study.optimize(_objective, n_trials=n_trials, timeout=timeout, n_jobs=n_jobs, gc_after_trial=gc_after_trial, callbacks=optuna_callbacks)
study.optimize(
_objective,
n_trials=n_trials,
timeout=timeout,
n_jobs=n_jobs,
gc_after_trial=gc_after_trial,
callbacks=optuna_callbacks,
)
if not study._is_multi_objective():
best_trial = study.best_trial
return BestRun(str(best_trial.number), best_trial.value, best_trial.params)