| { | |
| "model_type": "ddpm_unet", | |
| "image_size": 32, | |
| "in_channels": 3, | |
| "out_channels": 3, | |
| "base_channels": 128, | |
| "channel_multipliers": [1, 2, 2, 2], | |
| "attention_resolutions": [16], | |
| "num_res_blocks": 2, | |
| "dropout": 0.1, | |
| "num_heads": 4, | |
| "diffusion": { | |
| "timesteps": 1000, | |
| "beta_start": 0.0001, | |
| "beta_end": 0.02, | |
| "schedule": "cosine" | |
| }, | |
| "training": { | |
| "dataset": "CIFAR-10", | |
| "batch_size": 256, | |
| "learning_rate": 2e-4, | |
| "epochs": 300, | |
| "ema_decay": 0.9999, | |
| "mixed_precision": "bf16", | |
| "hardware": "RTX 5080 16GB" | |
| } | |
| } | |