big batch

newsearch.py

@@ -32,8 +32,8 @@ alpha_values = [0]
 # depths = [1, 2, 4, 8, 16]
 widths, depths = [512], [4]
 
-batch_size_values = [256]
-max_epochs_values = [50]
+batch_size_values = [1024]
+max_epochs_values = [250]
 seeds = list(range(21, 1992))
 optimizers = [
     # "Adagrad",