Commit 75530d88 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

improved settings layout + added AE settings

parent 0997b994
......@@ -40,15 +40,19 @@ use_last_checkpoint = False
kernel_heigth = 3
kernel_width = 3
kernel_depth = 3
kernel_classification = 1
; kernel_classification = 1
input_channels = 1
output_channels = 64
convolution_stride = 1
dropout = 0.2
pool_kernel_size = 2
; pool_kernel_size = 2
pool_stride = 2
up_mode = "upconv"
number_of_classes = 1
; ---> parameters for the ResNet CGAN
pool_kernel_size = 3
kernel_classification = 7
[MISC]
save_model_directory = "saved_models"
......
[MAPPING]
trained_model_path = "saved_models/finetuned_alldata.pth.tar"
trained_model_path = "saved_models/test24.pth.tar"
prediction_output_path = "test24_predictions"
data_directory = "/well/win-biobank/projects/imaging/data/data3/subjectsAll/"
mapping_data_file = "dMRI/autoptx_preproc/tractsNormSummed.nii.gz"
data_list = "datasets/test.txt"
prediction_output_path = "network_predictions"
brain_mask_path = "utils/MNI152_T1_2mm_brain_mask.nii.gz"
mean_mask_path = "utils/mean_dr_stage2.nii.gz"
scaling_factors = "datasets/scaling_factors.pkl"
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment