From 6d40d390979d34d2957414a3e761297eafc50ded Mon Sep 17 00:00:00 2001
From: "Michael Pilosov, PhD" <consistentbayes@gmail.com>
Date: Sat, 27 Jan 2024 09:38:43 +0000
Subject: [PATCH] save config to out

---
 newsearch.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/newsearch.py b/newsearch.py
index e7fc897..4d2a1fd 100644
--- a/newsearch.py
+++ b/newsearch.py
@@ -40,7 +40,7 @@ optimizers = [
 ]
 # Generate all possible combinations of hyperparameters
 all_params = [
-    (alpha, lr, bs, me, s, w)
+    (alpha, lr, bs, me, s, w, opt)
     for alpha in alpha_values
     for lr in learning_rate_values
     for bs in batch_size_values
@@ -82,12 +82,13 @@ python newmain.py fit \
 --lr_scheduler.init_args.cooldown 10 \
 --lr_scheduler.init_args.verbose true
 """
-
+    test_cmd = cmd + f" --print_config > out/config_v{idx:04d}.txt"
     # 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:
         # Run the command and wait for it to complete
+        subprocess.run(test_cmd, shell=True, check=True)
         subprocess.run(cmd, shell=True, check=True)
     except KeyboardInterrupt:
         print("Interrupted by user")