start search, looking promising

10 months ago · 2e3c44955f
2 changed files with 3 additions and 2 deletions
--- a/2
+++ b/2
@ -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
--- a/search.py
+++ b/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]