Browse Source

fixed alpha

new-sep-loss
Michael Pilosov 10 months ago
parent
commit
b5bb3fe3df
  1. 3
      search.py

3
search.py

@ -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]

Loading…
Cancel
Save