"...resnet50_tensorflow.git" did not exist on "46a373afe3d9e8850839c53d3f0fb64a9a12e9b2"
Unverified Commit 5fc3b5cb authored by Kai Fricke's avatar Kai Fricke Committed by GitHub
Browse files

Fix Tune progress_reporter kwarg (#7508)

parent dabc85d1
......@@ -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))
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
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