Hyunji

config

1 +config = {
2 + "debug" : False,
3 + "seed" : 0,
4 + "wandb.use" : False,
5 + "wandb.run_id" : None,
6 + "project" : "brain-age",
7 + "result_folder" : "result",
8 + "mode" : ["test", "train"],
9 +
10 + "data.name" : "brain_age",
11 + "data.root_path" : "**data root path",
12 + "data.train_csv" : "**data train csv",
13 + "data.test_csv" : "**data test csv",
14 + "data.valid_csv" : "**data valid csv",
15 + "data.train_num_sample" : -1,
16 + "data.frame_keep_style" : "random",
17 + "data.frame_keep_fraction" : 1,
18 + "data.frame_dim" : 1,
19 + "data.impute" : "drop",
20 +
21 + "model.name" : "regression",
22 +
23 + "model.arch.file" : "src/arch/brain_age_3d.py",
24 + "model.arch.lstm_feat_dim" : 2,
25 + "model.arch.lstm_latent_dim" : 128,
26 + "model.arch.attn_dim" : 32,
27 + "model.arch.attn_num_heads" : 1,
28 + "model.arch.attn_drop" : False,
29 + "model.arch.agg_fn" : "attention",
30 +
31 + "train.batch_size" : 8,
32 + "train.patience" : 100,
33 + "train.max_epoch" : 100,
34 + "train.optimizer" : "adam",
35 + "train.lr" : 1e-4,
36 + "train.weight_decay" : 1e-4,
37 + "train.save_strategy" : ["best", "last"],
38 + "train.log_every" : 100,
39 + "train.stopping_criteria" : "loss",
40 + "train.stopping_criteria_direction": "lower",
41 + "train.gradient_norm_clip" : -1,
42 +
43 + "test.batch_size" : 8,
44 + "test.eval_model" : "best",
45 +}