{
    "command": "train",
    "gpu_ids": [
        0
    ],
    "path_output": "BIDS_out",
    "model_name": "Model_SC_7T",
    "debugging": false,
    "object_detection_params": {
        "object_detection_path": null,
        "safety_factor": [
            1.0,
            1.0,
            1.0
        ],
        "gpu_ids": 0,
        "path_output": "BIDS_out"
    },
    "loader_parameters": {
        "path_data": [
            "BIDS_Train_Validation"
        ],
        "subject_selection": {
            "n": [],
            "metadata": [],
            "value": []
        },
        "target_suffix": [
            "_SC-manual"
        ],
        "extensions": [
            ".nii.gz"
        ],
        "roi_params": {
            "suffix": null,
            "slice_filter_roi": null
        },
        "contrast_params": {
            "training_validation": [
                "T2star"
            ],
            "testing": [
                "T2star"
            ],
            "balance": {}
        },
        "slice_filter_params": {
            "filter_empty_mask": false,
            "filter_empty_input": true
        },
        "slice_axis": "axial",
        "multichannel": false,
        "soft_gt": false
    },
    "split_dataset": {
        "fname_split": null,
        "random_seed": 6,
        "split_method": "participant_id",
        "data_testing": {
            "data_type": null,
            "data_value": []
        },
        "balance": null,
        "train_fraction": 0.6,
        "test_fraction": 0.2
    },
    "training_parameters": {
        "batch_size": 20,
        "loss": {
            "name": "DiceLoss"
        },
        "training_time": {
            "num_epochs": 1000,
            "early_stopping_patience": 50,
            "early_stopping_epsilon": 0.001
        },
        "scheduler": {
            "initial_lr": 0.001,
            "lr_scheduler": {
                "name": "CosineAnnealingLR",
                "base_lr": 1e-05,
                "max_lr": 0.01
            }
        },
        "balance_samples": {
            "applied": false,
            "type": "gt"
        },
        "mixup_alpha": null,
        "transfer_learning": {
            "retrain_model": null,
            "retrain_fraction": 1.0,
            "reset": true
        }
    },
    "default_model": {
        "name": "Unet",
        "dropout_rate": 0.3,
        "bn_momentum": 0.1,
        "depth": 3,
        "is_2d": true,
        "final_activation": "sigmoid"
    },
    "uncertainty": {
        "epistemic": false,
        "aleatoric": false,
        "n_it": 0
    },
    "postprocessing": {
        "remove_noise": {
            "thr": -1
        },
        "keep_largest": {},
        "binarize_prediction": {
            "thr": 0.5
        },
        "uncertainty": {
            "thr": -1,
            "suffix": "_unc-vox.nii.gz"
        },
        "fill_holes": {},
        "remove_small": {
            "unit": "vox",
            "thr": 3
        }
    },
    "evaluation_parameters": {
        "target_size": {
            "unit": "vox",
            "thr": [
                20,
                100
            ]
        },
        "overlap": {
            "unit": "vox",
            "thr": 3
        }
    },
    "transformation": {
        "Resample": {
            "wspace": 0.75,
            "hspace": 0.75,
            "dspace": 1
        },
        "CenterCrop": {
            "size": [
                128,
                128
            ]
        },
        "RandomAffine": {
            "degrees": 5,
            "scale": [
                0.1,
                0.1
            ],
            "translate": [
                0.03,
                0.03
            ],
            "applied_to": [
                "im",
                "gt"
            ],
            "dataset_type": [
                "training"
            ]
        },
        "ElasticTransform": {
            "alpha_range": [
                28.0,
                30.0
            ],
            "sigma_range": [
                3.5,
                4.5
            ],
            "p": 0.1,
            "applied_to": [
                "im",
                "gt"
            ],
            "dataset_type": [
                "training"
            ]
        },
        "NumpyToTensor": {},
        "NormalizeInstance": {
            "applied_to": [
                "im"
            ]
        }
    },
    "FiLMedUnet": {
        "applied": false,
        "metadata": "contrasts",
        "film_layers": [
            0,
            1,
            0,
            0,
            0,
            0,
            0,
            0,
            0,
            0
        ]
    },
    "Modified3DUNet": {
        "applied": false,
        "length_3D": [
            128,
            128,
            16
        ],
        "stride_3D": [
            128,
            128,
            16
        ],
        "attention": false,
        "n_filters": 8
    }
}