| 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 |
|
|
