Upload config.yaml

millionaid/config.yaml

@@ -0,0 +1,55 @@
+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