config.txt

{"training":{"device":"cuda:0",
             "trainsize": 8,
             "model_name":"VNet",
             "num_epochs":100,
             "k_follds_number":5,
             "batch_size" : 1,
             "learning_rate" : 0.0001,
             "momentum" : 0.99,
             "wd":0.01,
             "patience" : 30, 
             "delta" : 0.001,
             "step_size":30, 
             "gamma":0.1
                     },
 "loss":{"loss_name":"DCE", 
         "loss_type":"DCE", 
         "dce_w" : 1,
         "w0" : 1,
         "w1" : 1,
         "nll":"True"
                   },
 "predict":{"model_path":"../M_models/VNet_DCE/VNet_kf_DCE_1.pth"},
 "loaders" : {
        "train_patch": [112, 144, 64], 
        "train_stride": [112, 144, 64],
        "val_patch": [112, 144, 64], 
        "val_stride": [120, 150, 70],
        "dicom_path": "../../data/data_for_model/dicom",
        "mask_path": "../../data/data_for_model/mask", 
        "objects": ["LM_voi_segm_final","RM_voi_segm_final"], 
        "voi_shape": [120, 150, 70],
        "precrop": "True",
        "train_patient_ids": [1], 
        "val_patient_ids": [1,2,3,4,5,6,7,8,9,10], 
        "num_workers": 8, 
        "batch_size": 1,
        "train_dataset_size": 80,
    
        "transformer": {"train": 
        {"raw": [
             {"name": "Normalize"},
             {"name": "RandomFlip", 
                      "axes": [1],
                      "execution_probability":0.5},
             {"name": "RandomRotate", 
                      "axes": [[0,1],[0,2],[2,1]],
                      "angle_spectrum": 15,
                      "mode": "nearest"},
             {"name": "ElasticDeformation",
                  "spline_order": 2,
                  "alpha": 30,
                  "sigma": 5,
                  "execution_probability":0.5},
             {"name": "RandomContrast", 
                  "execution_probability": 0.5, 
                  "alpha":[1,3]}
                    ],
        "label":[
              {"name": "RandomFlip",
                      "axes": [1],
                      "execution_probability":0.5},
         {"name": "RandomRotate",
                  "axes": [[0,1],[0,2],[2,1]],
                  "angle_spectrum": 15,
                  "mode": "constant"}
                                ,
         {"name": "ElasticDeformation",
                  "spline_order": 0,
                  "alpha": 30,
                  "sigma": 5,
                  "execution_probability":0.5}
        ]
            },
        "test":{"raw":[{"name": "Normalize"}],
                "label": []}
        }
    }
}