Showing
1 changed file
with
45 additions
and
0 deletions
config/config.py
0 → 100644
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" : "/content/drive/MyDrive/2d-slice-set-networks-for-brain-age-master/data", | ||
12 | + "data.train_csv" : "data/train0609.csv", | ||
13 | + "data.test_csv" : "data/test0609.csv", | ||
14 | + "data.valid_csv" : "data/valid0609.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 | +} |
-
Please register or login to post a comment