[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 = True [TRAINING] experiment_name = "VA2-1" pre_trained_path = "saved_models/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] first_kernel_size = 9 first_convolution_stride = 1 input_channels = 1 output_channels = 32 max_number_channels = 128 kernel_size = 3 convolution_stride = 2 number_of_encoder_blocks = 2 transformer_blocks_stride = 1 number_of_transformer_blocks = 6 kernel_classification = 9 pool_kernel_size = 3 pool_stride = 2 dropout = 0 final_activation = 'tanh' number_of_classes = 1 custom_weight_reset_flag = False 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 [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"