start search, looking promising

makefile

@@ -4,7 +4,7 @@ lint:
 	flake8 --ignore E501,W503 .
 
 test:
-	python main.py --alpha 4 --lr 2e-4 --max_epochs 200
+	python main.py --alpha 2 --lr 2e-4 --max_epochs 200
 
 search:
 	python search.py

search.py

@@ -19,7 +19,8 @@ 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 = list(np.round(np.logspace(-5, -3, 41), 5))
+# learning_rate_values = list(np.round(np.logspace(-5, -3, 41), 5))
+learning_rate_values = [5e-4]
 batch_size_values = [64, 128]
 max_epochs_values = [500]