diff --git a/search.py b/search.py
index 7a9b3cc..d4413aa 100644
--- a/search.py
+++ b/search.py
@@ -18,10 +18,10 @@ 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(0, 2, 21), 4))
-learning_rate_values = [1e-3, 1e-4, 1e-5]
-batch_size_values = [128]
-max_epochs_values = [5000]
+alpha_values = list(np.round(np.linspace(0, 4, 41), 4))
+learning_rate_values = np.log(np.linspace(-5, -3, 11))
+batch_size_values = [64, 128]
+max_epochs_values = [500]
 
 # Generate all possible combinations of hyperparameters
 all_params = [