Skip to content

GitLab

Unverified Commit fab1de72 authored by Nolwenn Bernard's avatar Nolwenn Bernard Committed by GitHub
Browse files

Hyperparameter search reporting to W&B (#22440) 

Fixes #22429
parent 8d9c3836
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/integrations.py
View file @ fab1de72
...@@ -478,7 +478,7 @@ def run_hp_search_wandb(trainer, n_trials: int, direction: str, **kwargs) -> Bes ...@@ -478,7 +478,7 @@ def run_hp_search_wandb(trainer, n_trials: int, direction: str, **kwargs) -> Bes
break break
if not reporting_to_wandb: if not reporting_to_wandb:
trainer.add_callback(WandbCallback()) trainer.add_callback(WandbCallback())
trainer.args.report_to = "wandb" trainer.args.report_to = ["wandb"]
best_trial = {"run_id": None, "objective": None, "hyperparameters": None} best_trial = {"run_id": None, "objective": None, "hyperparameters": None}
sweep_id = kwargs.pop("sweep_id", None) sweep_id = kwargs.pop("sweep_id", None)
project = kwargs.pop("project", None) project = kwargs.pop("project", None)
......
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