some tweaks, ready to run at scale

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",