Browse Source

real run

plotting-unify
Michael Pilosov, PhD 9 months ago
parent
commit
822400191e
  1. 4
      newsearch.py

4
newsearch.py

@ -20,7 +20,7 @@ learning_rate_values = [1e-3]
alpha_values = [0, 0.1] alpha_values = [0, 0.1]
widths = [2**k for k in range(4, 13)] widths = [2**k for k in range(4, 13)]
depths = [1, 2, 4, 8, 16] depths = [1, 2, 4, 8, 16]
dropouts = [0, 0.25, 0.5] dropouts = [0, 0.5]
# widths, depths = [512], [4] # widths, depths = [512], [4]
batch_size_values = [256] batch_size_values = [256]
@ -97,7 +97,7 @@ cd ~/colors && python newmain.py fit \
studio = Studio("colors-animate-jobs") studio = Studio("colors-animate-jobs")
studio.install_plugin("jobs") studio.install_plugin("jobs")
job_plugin = studio.installed_plugins["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) job_plugin.run(cmd, machine=Machine.T4, name=job_name)
# Run the command and wait for it to complete # Run the command and wait for it to complete

Loading…
Cancel
Save