mmnist -> ct

nnaakkaaii · Jun 16, 2024 · 964809c · 964809c
1 parent bb219d7
commit 964809c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tuning/ct.py b/tuning/ct.py
@@ -278,7 +278,7 @@ def objective(trial):
     )
 
     result_dir = Path(
-        f"results/tuning/mmnist/{network_name}/{motion_encoder_name}/{trial.number}-{str(uuid4())[:8]}"
+        f"results/tuning/ct/{network_name}/{motion_encoder_name}/{trial.number}-{str(uuid4())[:8]}"
     )
     train_option = TrainExpOption(
         result_dir=result_dir,
@@ -319,7 +319,7 @@ def objective(trial):
     parser.add_argument("--rnn_name", type=str)
     args = parser.parse_args()
 
-    study_name = "mmnist"
+    study_name = "ct"
     if args.network_name is not None:
         assert args.network_name in [
             "hrdae3d",
@@ -345,11 +345,11 @@ def objective(trial):
         study_name += f"_{args.rnn_name}"
     study = optuna.create_study(
         study_name=study_name,
-        storage="sqlite:///results/tuning/mmnist/sqlite.db",
+        storage="sqlite:///results/tuning/ct/sqlite.db",
         load_if_exists=True,
     )
     study.optimize(objective, n_trials=500)
     print(study.best_params)
     print(study.best_value)
     print(study.best_trial)
-    study.trials_dataframe().to_csv("results/tuning/mmnist/trials.csv")
+    study.trials_dataframe().to_csv("results/tuning/ct/trials.csv")