Fix Tune progress_reporter kwarg (#7508)

2020-10-01 10:34:31 -04:00
parent dabc85d1ba
commit 5fc3b5cba4
1 changed files with 1 additions and 1 deletions
@@ -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"])