Browse Source

search

new-sep-loss
Michael Pilosov 11 months ago
parent
commit
deabe4dde6
  1. 8
      search.py

8
search.py

@ -18,10 +18,10 @@ 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, 2, 21), 4)) alpha_values = list(np.round(np.linspace(0, 4, 41), 4))
learning_rate_values = [1e-3, 1e-4, 1e-5] learning_rate_values = np.log(np.linspace(-5, -3, 11))
batch_size_values = [128] batch_size_values = [64, 128]
max_epochs_values = [5000] max_epochs_values = [500]
# Generate all possible combinations of hyperparameters # Generate all possible combinations of hyperparameters
all_params = [ all_params = [

Loading…
Cancel
Save