| seed: 2022 |
| data: |
| target: datasets.cifar10.CIFAR10 |
| params: |
| root: /data/fengxin/xuyifeng/data/CIFAR-10/ |
| img_size: 32 |
| img_channels: 3 |
| num_classes: 10 |
| 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 |
| - 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 |
| ema_decay: 0.9999 |
| ema_gradual: true |
| print_freq: 400 |
| save_freq: 10000 |
| sample_freq: 5000 |
| n_samples: 64 |
| optim: |
| target: torch.optim.Adam |
| params: |
| lr: 0.0002 |
|
|
