sample_config.json

{
  "model_type": "conditional_ddpm_camels",
  "in_channels": 1,
  "out_channels": 1,
  "image_size": 256,
  "label_dim": 2,
  "label_names": [
    "Omega_m",
    "sigma_8"
  ],
  "base_channels": 64,
  "channel_multipliers": [
    1,
    2,
    4,
    8
  ],
  "attention_levels": [
    2,
    3
  ],
  "dropout": 0.1,
  "timesteps": 1500,
  "beta_start": 0.0001,
  "beta_end": 0.02,
  "schedule_type": "linear",
  "ddim_steps_default": 50,
  "framework": "pytorch",
  "library_name": "pytorch",
  "training_meta": {
    "epochs": 100,
    "batch_size": 8,
    "lr": 0.0002,
    "ema_decay": 0.9999,
    "seed": 42
  },
  "label_mu": [
    0.3308129608631134,
    0.7831979990005493
  ],
  "label_std": [
    0.1140434592962265,
    0.12279357761144638
  ]
}