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