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