From 55f0f60bb8b39f6e43f431c30fe8828e4ff04b0d Mon Sep 17 00:00:00 2001 From: "Michael Pilosov, PhD" Date: Thu, 22 Feb 2024 07:49:54 +0000 Subject: [PATCH] prepare animations (supervised and not) --- newsearch.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) 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]