This commit is contained in:
Michael Pilosov, PhD 2024-03-03 22:46:13 +00:00
parent 2777c0d7ab
commit 01906c9f2a

View File

@ -24,7 +24,7 @@ dropouts = [0, 0.5]
# widths, depths = [512], [4]
batch_size_values = [64, 256]
max_epochs_values = [720] # at 12fps 720 frames = 60s
max_epochs_values = [720*3] # at 12fps 720 frames = 60s
seeds = list(range(21, 1992))
optimizers = [
# "Adagrad",
@ -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+1}"
job_name = f"colors-hour-20240303-{idx+1}"
job_plugin.run(cmd, machine=Machine.T4, name=job_name)
# Run the command and wait for it to complete