settings.ini 1.71 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 = True
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
8
9

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

[NETWORK]
Andrei Roibu's avatar
Andrei Roibu committed
27
first_kernel_size = 9
28
first_convolution_stride = 1
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
29
input_channels = 1
30
output_channels = 32
31
32
33
34
35
36
max_number_channels = 128
kernel_size = 3
convolution_stride = 2
number_of_encoder_blocks = 2
transformer_blocks_stride = 1
number_of_transformer_blocks = 6
Andrei Roibu's avatar
Andrei Roibu committed
37
kernel_classification = 9
38
pool_kernel_size = 3
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
39
pool_stride = 2
40
dropout = 0
Andrei Roibu's avatar
Andrei Roibu committed
41
final_activation = 'tanh'
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
42
number_of_classes = 1
43
custom_weight_reset_flag = False
44
45
46
47
48
cross_domain_flag = False
cross_domain_x2x_flag = False
cross_domain_y2y_flag = False
cross_domain_x2y_flag = False
number_of_feature_mapping_blocks = 1
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
49
50
51
52

[MISC]
save_model_directory = "saved_models"
logs_directory = "logs"
53
checkpoint_directory = "checkpoints"
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
54
experiments_directory = "experiments"
55
device = 0
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
56

57
[EVALUATION]
Andrei-Claudiu Roibu's avatar
Andrei-Claudiu Roibu committed
58
59
60
61
62
63
64
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"
65