Hugging Face's logo

xyfJASON
/
Diffusion-Models-Implementations

TensorBoard
Model card Files Metrics Community
xyfJASON commited on
Commit
7c06239
•
1 Parent(s): 3c6426b

Update config file

Browse files
Files changed (2) hide show
  1. ddpm_celebahq/config-2023-03-09-21-50-59.yaml +46 -52
  2. ddpm_celebahq/config-2023-03-14-11-16-58.yaml +46 -42
ddpm_celebahq/config-2023-03-09-21-50-59.yaml CHANGED
@@ -1,64 +1,58 @@
 
1
  data:
2
- dataroot: /data/fengxin/xuyifeng/data/CelebA-HQ/
 
 
 
3
  img_channels: 3
4
- img_size: 256
5
- name: CelebA-HQ
6
  dataloader:
7
- micro_batch: 8
8
  num_workers: 4
9
  pin_memory: true
10
  prefetch_factor: 2
11
- diffusion:
12
- beta_end: 0.02
13
- beta_schedule: linear
14
- beta_start: 0.0001
15
- objective: pred_eps
16
- total_steps: 1000
17
- var_type: fixed_small
18
  model:
19
- dim: 128
20
- dim_mults:
21
- - 1
22
- - 1
23
- - 2
24
- - 2
25
- - 4
26
- - 4
27
- dropout: 0.0
28
- ema_decay: 0.9999
29
- ema_gradual: true
30
- in_channels: 3
31
- n_heads: 1
32
- num_res_blocks: 2
33
- out_channels: 3
34
- type: unet
35
- use_attn:
36
- - false
37
- - false
38
- - false
39
- - false
40
- - true
41
- - false
42
- sample:
43
- load_ema: true
44
- micro_batch: 50
45
- mode: sample
46
- n_denoise: 20
47
- n_progressive: 20
48
- n_samples: null
49
- save_dir: null
50
- skip_steps: null
51
- weights: null
52
- seed: 2022
53
  train:
 
54
  batch_size: 64
 
55
  clip_grad_norm: 1.0
56
- n_samples: 36
57
- n_steps: 500000
58
- optim:
59
- lr: 2.0e-05
60
- type: Adam
61
  print_freq: 400
62
- resume: null
63
- sample_freq: 5000
64
  save_freq: 10000
 
 
 
 
 
 
 
1
+ seed: 2022
2
  data:
3
+ target: datasets.celebahq.CelebAHQ
4
+ params:
5
+ root: /data/fengxin/xuyifeng/data/CelebA-HQ/
6
+ img_size: 256
7
  img_channels: 3
 
 
8
  dataloader:
 
9
  num_workers: 4
10
  pin_memory: true
11
  prefetch_factor: 2
 
 
 
 
 
 
 
12
  model:
13
+ target: models.unet.UNet
14
+ params:
15
+ in_channels: 3
16
+ out_channels: 3
17
+ dim: 128
18
+ dim_mults:
19
+ - 1
20
+ - 1
21
+ - 2
22
+ - 2
23
+ - 4
24
+ - 4
25
+ use_attn:
26
+ - false
27
+ - false
28
+ - false
29
+ - false
30
+ - true
31
+ - false
32
+ num_res_blocks: 2
33
+ n_heads: 1
34
+ dropout: 0.0
35
+ diffusion:
36
+ target: diffusions.ddpm.DDPM
37
+ params:
38
+ total_steps: 1000
39
+ beta_schedule: linear
40
+ beta_start: 0.0001
41
+ beta_end: 0.02
42
+ objective: pred_eps
43
+ var_type: fixed_small
 
 
 
44
  train:
45
+ n_steps: 500000
46
  batch_size: 64
47
+ micro_batch: 8
48
  clip_grad_norm: 1.0
49
+ ema_decay: 0.9999
50
+ ema_gradual: true
 
 
 
51
  print_freq: 400
 
 
52
  save_freq: 10000
53
+ sample_freq: 5000
54
+ n_samples: 36
55
+ optim:
56
+ target: torch.optim.Adam
57
+ params:
58
+ lr: 2.0e-05
ddpm_celebahq/config-2023-03-14-11-16-58.yaml CHANGED
@@ -1,54 +1,58 @@
 
1
  data:
2
- dataroot: /amax/xyf/data/CelebA-HQ/
 
 
 
3
  img_channels: 3
4
- img_size: 256
5
- name: CelebA-HQ
6
  dataloader:
7
- micro_batch: 8
8
  num_workers: 4
9
  pin_memory: true
10
  prefetch_factor: 2
11
- diffusion:
12
- beta_end: 0.02
13
- beta_schedule: linear
14
- beta_start: 0.0001
15
- objective: pred_eps
16
- total_steps: 1000
17
- var_type: fixed_small
18
  model:
19
- dim: 128
20
- dim_mults:
21
- - 1
22
- - 1
23
- - 2
24
- - 2
25
- - 4
26
- - 4
27
- dropout: 0.0
28
- ema_decay: 0.9999
29
- ema_gradual: true
30
- in_channels: 3
31
- n_heads: 1
32
- num_res_blocks: 2
33
- out_channels: 3
34
- type: unet
35
- use_attn:
36
- - false
37
- - false
38
- - false
39
- - false
40
- - true
41
- - false
42
- seed: 2022
 
 
 
 
 
 
 
43
  train:
 
44
  batch_size: 64
 
45
  clip_grad_norm: 1.0
46
- n_samples: 36
47
- n_steps: 500000
48
- optim:
49
- lr: 2.0e-05
50
- type: Adam
51
  print_freq: 400
52
- resume: ./runs/ddpm_celebahq/ckpt/step349999/
53
- sample_freq: 5000
54
  save_freq: 10000
 
 
 
 
 
 
 
1
+ seed: 2022
2
  data:
3
+ target: datasets.celebahq.CelebAHQ
4
+ params:
5
+ root: /data/fengxin/xuyifeng/data/CelebA-HQ/
6
+ img_size: 256
7
  img_channels: 3
 
 
8
  dataloader:
 
9
  num_workers: 4
10
  pin_memory: true
11
  prefetch_factor: 2
 
 
 
 
 
 
 
12
  model:
13
+ target: models.unet.UNet
14
+ params:
15
+ in_channels: 3
16
+ out_channels: 3
17
+ dim: 128
18
+ dim_mults:
19
+ - 1
20
+ - 1
21
+ - 2
22
+ - 2
23
+ - 4
24
+ - 4
25
+ use_attn:
26
+ - false
27
+ - false
28
+ - false
29
+ - false
30
+ - true
31
+ - false
32
+ num_res_blocks: 2
33
+ n_heads: 1
34
+ dropout: 0.0
35
+ diffusion:
36
+ target: diffusions.ddpm.DDPM
37
+ params:
38
+ total_steps: 1000
39
+ beta_schedule: linear
40
+ beta_start: 0.0001
41
+ beta_end: 0.02
42
+ objective: pred_eps
43
+ var_type: fixed_small
44
  train:
45
+ n_steps: 500000
46
  batch_size: 64
47
+ micro_batch: 8
48
  clip_grad_norm: 1.0
49
+ ema_decay: 0.9999
50
+ ema_gradual: true
 
 
 
51
  print_freq: 400
 
 
52
  save_freq: 10000
53
+ sample_freq: 5000
54
+ n_samples: 36
55
+ optim:
56
+ target: torch.optim.Adam
57
+ params:
58
+ lr: 2.0e-05