diff --git a/image_ref/main_ray.py b/image_ref/main_ray.py index 81564a83a014c195ff5f6577014dca7d8d5c4f18..d4eac83711982e631e421a8dfe8b2b0cd3896847 100644 --- a/image_ref/main_ray.py +++ b/image_ref/main_ray.py @@ -231,7 +231,7 @@ def main(args, gpus_per_trial=1): resources={"cpu": 20, "gpu": gpus_per_trial} ), tune_config=tune.TuneConfig( - time_budget_s=3600 * 23.5, + time_budget_s=3600 * 19.5, search_alg=algo, scheduler=scheduler, num_samples=50, @@ -240,7 +240,7 @@ def main(args, gpus_per_trial=1): ), run_config=RunConfig(storage_path="/lustre/fswork/projects/rech/bun/ucg81ws/these/pseudo_image/image_ref/ray_results_test", - name="test_experiment_no_scheduler" + name="base_experiment" ), param_space=config