settings.ini 1.52 KB
Newer Older
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
1
[DATA]
2
data_folder_name = "datasets"
3
4
5
6
input_data_train = "input_data_train.h5"
target_data_train = "target_data_train.h5"
input_data_validation = "input_data_validation.h5"
target_data_validation = "target_data_validation.h5"
7
crop_flag = False
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
8
9

[TRAINING]
10
11
12
experiment_name = "VA2-1"
pre_trained_path = "saved_models/VA2-1.pth.tar"
final_model_output_file = "VA2-1.pth.tar"
13
14
training_batch_size = 3
validation_batch_size = 3
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
15
use_pre_trained = False
Andrei Roibu's avatar
Andrei Roibu committed
16
learning_rate = 1e-5
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
17
18
19
optimizer_beta = (0.9, 0.999)
optimizer_epsilon = 1e-8
optimizer_weigth_decay = 1e-5
20
number_of_epochs = 10
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
21
loss_log_period = 50
22
learning_rate_scheduler_step_size = 6
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
23
learning_rate_scheduler_gamma = 1e-1
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
24
use_last_checkpoint = False
Andrei Roibu's avatar
Andrei Roibu committed
25
adam_w_flag = False
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
26
27

[NETWORK]
28
29
30
kernel_heigth = 3
kernel_width = 3
kernel_depth = 3
31
kernel_classification = 7
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
32
input_channels = 1
33
output_channels = 32
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
34
convolution_stride = 1
35
dropout = 0
36
pool_kernel_size = 3
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
37
38
pool_stride = 2
up_mode = "upconv"
39
final_activation = 'tanh'
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
40
number_of_classes = 1
41
number_of_transformer_blocks = 6
42
custom_weight_reset_flag = False
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
43
44
45
46

[MISC]
save_model_directory = "saved_models"
logs_directory = "logs"
47
checkpoint_directory = "checkpoints"
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
48
experiments_directory = "experiments"
49
device = 0
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
50

51
[EVALUATION]
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
52
53
54
55
56
57
58
trained_model_path = "saved_models/model.pth.tar"
data_directory = "/well/win-biobank/projects/imaging/data/data3/subjectsAll/"
data_file = "/dMRI/autoptx_preproc/tractsNormSummed.nii.gz"
output_targets = "/fMRI/rfMRI_25.dr/dr_stage2.nii.gz"
data_list = "test.txt"
orientation = "coronal"
saved_predictions_directory = "predictions"
59