Update config file

ddpm_cifar10/config-2023-03-09-14-06-16.yaml

@@ -1,51 +1,55 @@
+seed: 2022
 data:
-  dataroot: /data/fengxin/xuyifeng/data/CIFAR-10/
+  target: datasets.cifar10.CIFAR10
+  params:
+    root: /data/fengxin/xuyifeng/data/CIFAR-10/
+    img_size: 32
   img_channels: 3
-  img_size: 32
-  name: CIFAR-10
   num_classes: 10
 dataloader:
-  micro_batch: 0
   num_workers: 4
   pin_memory: true
   prefetch_factor: 2
-diffusion:
-  beta_end: 0.02
-  beta_schedule: linear
-  beta_start: 0.0001
-  objective: pred_eps
-  total_steps: 1000
-  var_type: fixed_large
 model:
-  dim: 128
-  dim_mults:
-  - 1
-  - 2
-  - 2
-  - 2
-  dropout: 0.1
-  ema_decay: 0.9999
-  ema_gradual: true
-  in_channels: 3
-  n_heads: 1
-  num_res_blocks: 2
-  out_channels: 3
-  type: unet
-  use_attn:
-  - false
-  - true
-  - false
-  - false
-seed: 2022
+  target: models.unet.UNet
+  params:
+    in_channels: 3
+    out_channels: 3
+    dim: 128
+    dim_mults:
+    - 1
+    - 2
+    - 2
+    - 2
+    use_attn:
+    - false
+    - true
+    - false
+    - false
+    num_res_blocks: 2
+    n_heads: 1
+    dropout: 0.1
+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_large
 train:
+  n_steps: 800000
   batch_size: 128
+  micro_batch: 0
   clip_grad_norm: 1.0
-  n_samples: 64
-  n_steps: 800000
-  optim:
-    lr: 0.0002
-    type: Adam
+  ema_decay: 0.9999
+  ema_gradual: true
   print_freq: 400
-  resume: null
-  sample_freq: 5000
   save_freq: 10000
+  sample_freq: 5000
+  n_samples: 64
+  optim:
+    target: torch.optim.Adam
+    params:
+      lr: 0.0002