From b5bb3fe3df633fa5b76efab82a1f566ad3bb417d Mon Sep 17 00:00:00 2001 From: Michael Pilosov Date: Mon, 15 Jan 2024 05:32:17 +0000 Subject: [PATCH] fixed alpha --- search.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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]