millionaid/config.yaml

experiment:
  experiment_name: million_aid_05-24-2024_18-17-23-687720
  exp_dir: under what experiment directory to put the save dir
  save_dir: directory name, where all ckpt and images will be saved
wandb:
  project: projectname
  entity: your user name or entity
  mode: online
model:
  _target_: lightning_uq_box.uq_methods.DDPM
  diffusion_model:
    _target_: denoising_diffusion_pytorch.denoising_diffusion_pytorch.GaussianDiffusion
    model:
      _target_: denoising_diffusion_pytorch.denoising_diffusion_pytorch.Unet
      dim: 64
      dim_mults:
      - 1
      - 2
      - 4
      - 8
      - 16
      - 32
      channels: 3
      out_dim: 3
      self_condition: true
    image_size: 224
    timesteps: 500
    sampling_timesteps: 250
    beta_schedule: cosine
  log_samples_every_n_steps: 1000
  optimizer:
    _target_: torch.optim.Adam
    _partial_: true
    lr: 0.0001
datamodule:
  _target_: millionaid_dm.MillionAIDDataModule
  root: "path_to_dataset"
  split: test
  batch_size: 16
  num_workers: 4
trainer:
  _target_: lightning.pytorch.Trainer
  accelerator: gpu
  strategy: ddp
  accumulate_grad_batches: 4
  devices:
  - 4
  - 5
  - 6
  - 7
  min_epochs: 800
  max_epochs: 1000
  log_every_n_steps: 50
  enable_progress_bar: true
  enable_checkpointing: true