seed: 2022
data:
  target: datasets.celebahq.CelebAHQ
  params:
    root: /data/fengxin/xuyifeng/data/CelebA-HQ/
    img_size: 256
  img_channels: 3
dataloader:
  num_workers: 4
  pin_memory: true
  prefetch_factor: 2
model:
  target: models.unet.UNet
  params:
    in_channels: 3
    out_channels: 3
    dim: 128
    dim_mults:
    - 1
    - 1
    - 2
    - 2
    - 4
    - 4
    use_attn:
    - false
    - false
    - false
    - false
    - true
    - false
    num_res_blocks: 2
    n_heads: 1
    dropout: 0.0
diffusion:
  target: diffusions.ddpm.DDPM
  params:
    total_steps: 1000
    beta_schedule: linear
    beta_start: 0.0001
    beta_end: 0.02
    objective: pred_eps
    var_type: fixed_small
train:
  n_steps: 500000
  batch_size: 64
  micro_batch: 8
  clip_grad_norm: 1.0
  ema_decay: 0.9999
  ema_gradual: true
  print_freq: 400
  save_freq: 10000
  sample_freq: 5000
  n_samples: 36
  optim:
    target: torch.optim.Adam
    params:
      lr: 2.0e-05