diff --git a/search.py b/search.py index d4413aa..4beff70 100644 --- a/search.py +++ b/search.py @@ -19,7 +19,7 @@ job_plugin = studio.installed_plugins["jobs"] # Define the ranges or sets of values for each hyperparameter alpha_values = list(np.round(np.linspace(0, 4, 41), 4)) -learning_rate_values = np.log(np.linspace(-5, -3, 11)) +learning_rate_values = list(np.round(np.logspace(-5, -3, 41), 5)) batch_size_values = [64, 128] max_epochs_values = [500] @@ -39,7 +39,7 @@ search_params = sample(all_params, min(NUM_JOBS, len(all_params))) for idx, params in enumerate(search_params): a, lr, bs, me = params cmd = f"cd ~/colors && python main.py --alpha {a} --lr {lr} --bs {bs} --max_epochs {me}" - job_name = f"color2_{bs}_{a}_{lr:2.2e}" + # job_name = f"color2_{bs}_{a}_{lr:2.2e}" # job_plugin.run(cmd, machine=Machine.T4, name=job_name) print(f"Running {params}: {cmd}") try: