Hyperparameter search reporting to W&B (#22440)

Fixes #22429

Hyperparameter search reporting to W&B (#22440)
Fixes #22429
fab1de72 · Nolwenn Bernard · GitHub · 8d9c3836 · fab1de72
Unverified Commit fab1de72 authored Mar 29, 2023 by Nolwenn Bernard Committed by GitHub Mar 29, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/integrations.py src/transformers/integrations.py +1 -1

No files found.
--- a/src/transformers/integrations.py
+++ b/src/transformers/integrations.py
@@ -478,7 +478,7 @@ def run_hp_search_wandb(trainer, n_trials: int, direction: str, **kwargs) -> Bes
            break
    if not reporting_to_wandb:
        trainer.add_callback(WandbCallback())
-    trainer.args.report_to = "wandb"
+    trainer.args.report_to = ["wandb"]
    best_trial = {"run_id": None, "objective": None, "hyperparameters": None}
    sweep_id = kwargs.pop("sweep_id", None)
    project = kwargs.pop("project", None)