Commit 29266e50 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

updated settings for debugging

parent 9338b510
......@@ -12,10 +12,10 @@ validation_data_file = "dMRI/autoptx_preproc/tractsNormSummed.nii.gz"
validation_target_file = "fMRI/rfMRI_25.dr/dr_stage2.nii.gz"
[TRAINING]
training_batch_size = 5
validation_batch_size = 5
training_batch_size = 2
validation_batch_size = 2
use_pre_trained = False
pre_trained_path = "saved_models/preTrained.pth.tar"
pre_trained_path = "saved_models/finetuned_alldata.pth.tar"
experiment_name = "experiment_name"
learning_rate = 1e-4
optimizer_beta = (0.9, 0.999)
......
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