fixed alpha

11 months ago · b5bb3fe3df
1 changed files with 2 additions and 1 deletions
--- 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]