diff --git a/search.py b/search.py index 1b6eaa7..757a048 100644 --- a/search.py +++ b/search.py @@ -18,8 +18,9 @@ job_plugin = studio.installed_plugins["jobs"] # do a sweep over learning rates # Define the ranges or sets of values for each hyperparameter -alpha_values = list(np.round(np.linspace(2, 4, 21), 4)) +# alpha_values = list(np.round(np.linspace(2, 4, 21), 4)) # learning_rate_values = list(np.round(np.logspace(-5, -3, 41), 5)) +alpha_values = [1] learning_rate_values = [5e-4] batch_size_values = [128] max_epochs_values = [500]