diff --git a/newsearch.py b/newsearch.py index e7c8054..4afd100 100644 --- a/newsearch.py +++ b/newsearch.py @@ -20,7 +20,7 @@ learning_rate_values = [1e-3] alpha_values = [0, 0.1] widths = [2**k for k in range(4, 13)] depths = [1, 2, 4, 8, 16] -dropouts = [0, 0.25, 0.5] +dropouts = [0, 0.5] # widths, depths = [512], [4] batch_size_values = [256] @@ -97,7 +97,7 @@ cd ~/colors && python newmain.py fit \ studio = Studio("colors-animate-jobs") studio.install_plugin("jobs") job_plugin = studio.installed_plugins["jobs"] - job_name = f"colors-animate-20240303-{idx}" + job_name = f"colors-animate-20240303-{idx+1}" job_plugin.run(cmd, machine=Machine.T4, name=job_name) # Run the command and wait for it to complete