Skip to content

GitLab

"comfy/git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "274dff3257bc9118e00a75d8f90b472e945c5c09"
Unverified Commit 5fc3b5cb authored by Kai Fricke's avatar Kai Fricke Committed by GitHub
Browse files

Fix Tune progress_reporter kwarg (#7508)

parent dabc85d1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/integrations.py
View file @ 5fc3b5cb
...@@ -141,7 +141,7 @@ def run_hp_search_ray(trainer, n_trials: int, direction: str, **kwargs) -> BestR ...@@ -141,7 +141,7 @@ def run_hp_search_ray(trainer, n_trials: int, direction: str, **kwargs) -> BestR
num_gpus_per_trial = int(math.ceil(num_gpus_per_trial / n_jobs)) num_gpus_per_trial = int(math.ceil(num_gpus_per_trial / n_jobs))
kwargs["resources_per_trial"] = {"gpu": num_gpus_per_trial} kwargs["resources_per_trial"] = {"gpu": num_gpus_per_trial}
if "reporter" not in kwargs: if "progress_reporter" not in kwargs:
from ray.tune import CLIReporter from ray.tune import CLIReporter
kwargs["progress_reporter"] = CLIReporter(metric_columns=["objective"]) kwargs["progress_reporter"] = CLIReporter(metric_columns=["objective"])
......
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