| { | |
| "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 | |
| } | |