Showing
1 changed file
with
0 additions
and
45 deletions
config/config.py
deleted
100644 → 0
| 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