|
@ -18,7 +18,7 @@ job_plugin = studio.installed_plugins["jobs"] |
|
|
# do a sweep over learning rates |
|
|
# do a sweep over learning rates |
|
|
|
|
|
|
|
|
# Define the ranges or sets of values for each hyperparameter |
|
|
# Define the ranges or sets of values for each hyperparameter |
|
|
alpha_values = list(np.round(np.linspace(0, 4, 41), 4)) |
|
|
alpha_values = list(np.round(np.linspace(2, 6, 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] |
|
|
learning_rate_values = [5e-4] |
|
|
batch_size_values = [64, 128] |
|
|
batch_size_values = [64, 128] |
|
|