args.json

{
  "label_dim": 6,
  "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",
  "epochs": 200,
  "batch_size": 8,
  "lr": 0.0002,
  "ema_decay": 0.9999,
  "num_workers": 4,
  "early_stop_patience": 100,
  "use_amp": true,
  "data_dir": "<DDPM_ROOT>/data/LH_data/params_6",
  "normalize_labels": true,
  "output_dir": "outputs_conditional_6param",
  "resume": "",
  "resume_refresh_scheduler": false,
  "sample_every": 10,
  "use_ddim": true,
  "ddim_steps": 50,
  "use_wandb": false,
  "wandb_project": "ddpm_cosmology",
  "wandb_entity": "",
  "wandb_run_name": ""
}