diff --git a/newsearch.py b/newsearch.py index 4132a5d..a4b2783 100644 --- a/newsearch.py +++ b/newsearch.py @@ -26,10 +26,11 @@ NUM_JOBS = 100 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. -widths = [2**k for k in range(4, 13)] -depths = [1, 2, 4, 8, 16] -# widths, depths = [128, 256], [4, 8] +# alpha_values = [0, .25, 0.5, 0.75, 1] # alpha = 0 is unsupervised. alpha = 1 is supervised. +alpha_values = [1.0] +# widths = [2**k for k in range(4, 13)] +# depths = [1, 2, 4, 8, 16] +widths, depths = [64, 128, 256], [4, 8] batch_size_values = [256] max_epochs_values = [10]