Blame for code/config.json

{
    "job_type": "standard",
    "GLOBAL_ROUNDS": 2,
    "LOCAL_ROUNDS": 300,
    "VALIDATE_EVERY": 25,
    "LR_START": 0.001,
    "LR_END": 5e-05,
    "LR_DECAY_EPOCHS": 200,
    "GRAD_CLIP_NORM": 1.0,
    "VAL_SPLIT": 0.2,
    "BATCH_SIZE": 6,
    "SAVE_DEBUG_SAMPLES": true,
    "spatial_shape": [
        128,
        128
    ],
    "net_config": {
        "spatial_dims": 2,
        "diffusion_model": {
            "in_channels": 1,
            "out_channels": 1,
            "channels": [
                64,
                128,
                192,
                256
            ],
            "num_res_blocks": [
                2,
                2,
                2,
                2
            ],
            "attention_levels": [
                false,
                false,
                true,
                true
            ]
        }
    },
    "DATALOADER_WORKERS": 8
}