Skip to content

Commit

Permalink
update study name
Browse files Browse the repository at this point in the history
  • Loading branch information
nnaakkaaii committed Jun 16, 2024
1 parent 77a7bf8 commit 3f6cdc0
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions tuning/mmnist.py
Original file line number Diff line number Diff line change
Expand Up @@ -249,13 +249,20 @@ def objective(trial):
parser.add_argument("--rnn_name", type=str)
args = parser.parse_args()

study_name = "mmnist"
if hasattr(args, "network_name"):
study_name += f"_{args.network_name}"
if hasattr(args, "motion_encoder_name"):
study_name += f"_{args.motion_encoder_name}"
if hasattr(args, "rnn_name"):
study_name += f"_{args.rnn_name}"
study = optuna.create_study(
study_name="mmnist",
study_name=study_name,
storage="sqlite:///results/tuning/mmnist/sqlite.db",
load_if_exists=True,
)
study.optimize(objective, n_trials=100)
print(study.best_params)
print(study.best_value)
print(study.best_trial)
study.trials_dataframe().to_csv("result/tuning/mmnist/trials.csv")
study.trials_dataframe().to_csv("results/tuning/mmnist/trials.csv")

0 comments on commit 3f6cdc0

Please sign in to comment.