[DATA] data_folder_name = "datasets" 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" crop_flag = False [TRAINING] experiment_name = "VA2-1" pre_trained_path = "saved_models/VA2-1.pth.tar" final_model_output_file = "VA2-1.pth.tar" training_batch_size = 3 validation_batch_size = 3 use_pre_trained = False learning_rate = 1e-5 optimizer_beta = (0.9, 0.999) optimizer_epsilon = 1e-8 optimizer_weigth_decay = 1e-5 number_of_epochs = 10 loss_log_period = 50 learning_rate_scheduler_step_size = 6 learning_rate_scheduler_gamma = 1e-1 use_last_checkpoint = False adam_w_flag = False [NETWORK] kernel_heigth = 3 kernel_width = 3 kernel_depth = 3 kernel_classification = 7 input_channels = 1 output_channels = 32 convolution_stride = 1 dropout = 0 pool_kernel_size = 3 pool_stride = 2 up_mode = "upconv" final_activation = 'tanh' number_of_classes = 1 number_of_transformer_blocks = 6 custom_weight_reset_flag = False [MISC] save_model_directory = "saved_models" logs_directory = "logs" checkpoint_directory = "checkpoints" experiments_directory = "experiments" device = 0 [EVALUATION] 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"