diff --git a/newsearch.py b/newsearch.py index 222c164..56e10d7 100644 --- a/newsearch.py +++ b/newsearch.py @@ -8,7 +8,7 @@ from lightning_sdk import Machine, Studio # noqa: F401 # consistency of randomly sampled experiments. seed(19920921) -NUM_JOBS = 10 +NUM_JOBS = 20 # Define the ranges or sets of values for each hyperparameter # alpha_values = list(np.round(np.linspace(2, 4, 21), 4)) @@ -19,12 +19,12 @@ learning_rate_values = [1e-3] # alpha_values = [0, .25, 0.5, 0.75, 1] # alpha = 0 is unsupervised. alpha = 1 is supervised. alpha_values = [0, 0.1] widths = [2**k for k in range(4, 13)] -depths = [1, 2, 4, 8, 16] +depths = [1, 2, 4, 8] dropouts = [0, 0.5] # widths, depths = [512], [4] -batch_size_values = [256] -max_epochs_values = [600] # at 12 fps, around 35s +batch_size_values = [64, 256] +max_epochs_values = [720] # at 12fps 720 frames = 60s seeds = list(range(21, 1992)) optimizers = [ # "Adagrad",