diff --git a/newsearch.py b/newsearch.py index 7a1b289..56a1823 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 = 1 +NUM_JOBS = 100 # reference to the current studio # if you run outside of Lightning, you can pass the Studio name @@ -27,10 +27,10 @@ learning_rate_values = [1e-3] # learning_rate_values = [5e-4] # alpha_values = [0, .25, 0.5, 0.75, 1] # alpha = 0 is unsupervised. alpha = 1 is supervised. -alpha_values = [0.1] -# widths = [2**k for k in range(4, 13)] -# depths = [1, 2, 4, 8, 16] -widths, depths = [512], [4] +alpha_values = [0, 0.1] +widths = [2**k for k in range(4, 13)] +depths = [1, 2, 4, 8, 16] +# widths, depths = [512], [4] batch_size_values = [256] max_epochs_values = [100]