really prepare for jobs

newsearch.py

@@ -10,15 +10,6 @@ seed(19920921)
 
 NUM_JOBS = 5
 
-# reference to the current studio
-# if you run outside of Lightning, you can pass the Studio name
-# studio = Studio()
-
-# use the jobs plugin
-# studio.install_plugin("jobs")
-# job_plugin = studio.installed_plugins["jobs"]
-
-# do a sweep over learning rates
 
 # Define the ranges or sets of values for each hyperparameter
 # alpha_values = list(np.round(np.linspace(2, 4, 21), 4))
@@ -81,7 +72,7 @@ cd ~/colors && python newmain.py fit \
 --model.bias true \
 --model.loop true \
 --model.transform tanh \
---model.dropout 0 \
+--model.dropout 0.5 \
 --trainer.min_epochs 10 \
 --trainer.max_epochs {me} \
 --trainer.log_every_n_steps 3 \
@@ -100,13 +91,16 @@ cd ~/colors && python newmain.py fit \
 # --lr_scheduler.init_args.cooldown 10 \
 # --lr_scheduler.init_args.verbose true
 """
-    # job_name = f"color2_{bs}_{a}_{lr:2.2e}"
-    # job_plugin.run(cmd, machine=Machine.T4, name=job_name)
     print(f"Running {params}: {cmd}")
     try:
+        studio = Studio("colors-animate-jobs")
+        studio.install_plugin("jobs")
+        job_plugin = studio.installed_plugins["jobs"]
+        job_name = f"color-animate-{idx}-{s}"
+        job_plugin.run(cmd, machine=Machine.T4, name=job_name)
+
         # Run the command and wait for it to complete
-        # subprocess.run(test_cmd, shell=True, check=True)
-        subprocess.run(cmd, shell=True, check=True)
+        # subprocess.run(cmd, shell=True, check=True)
     except KeyboardInterrupt:
         print("Interrupted by user")
         sys.exit(1)