|
@ -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 |
|
|