Hugging Face's logo

xyfJASON
/
Diffusion-Models-Implementations

TensorBoard
Model card Files
xet
Metrics Community
xyfJASON commited on
Commit
dd99824
·
1 Parent(s): 7c06239

Upload ddpm_cfg_cifar10 checkpoints and training logs

Browse files
Files changed (8) hide show
  1. ddpm_cfg_cifar10/ckpt/step799999/ema.pt +3 -0
  2. ddpm_cfg_cifar10/ckpt/step799999/meta.pt +3 -0
  3. ddpm_cfg_cifar10/ckpt/step799999/model.pt +3 -0
  4. ddpm_cfg_cifar10/ckpt/step799999/optimizer.pt +3 -0
  5. ddpm_cfg_cifar10/config-2023-05-06-19-03-22.yaml +58 -0
  6. ddpm_cfg_cifar10/output-2023-05-06-19-03-22.log +0 -0
  7. ddpm_cfg_cifar10/samples.zip +3 -0
  8. ddpm_cfg_cifar10/tensorboard/events.out.tfevents.1683371004.ubuntu-WS-C621E-SAGE-Series.1587378.0 +3 -0
ddpm_cfg_cifar10/ckpt/step799999/ema.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:729bd89018bf1d2d1f92d5de3f7982a22ca0f04546304db1e34c1a94001a0d3f
3
+ size 176845547
ddpm_cfg_cifar10/ckpt/step799999/meta.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7fc389681c8eb2922fbeabf0c843cedbad10aaaa30455d0d2b53d1e86ad206b
3
+ size 425
ddpm_cfg_cifar10/ckpt/step799999/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e994857389504ef03b4758693f5b363030fd940e400cee4995f53f59a665234
3
+ size 176868203
ddpm_cfg_cifar10/ckpt/step799999/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a1f495f8e89e9f1c8dc279712073837a5bb11b78519f6174b9e65049a08d241
3
+ size 353790302
ddpm_cfg_cifar10/config-2023-05-06-19-03-22.yaml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ seed: 2022
2
+ data:
3
+ target: datasets.cifar10.CIFAR10
4
+ params:
5
+ root: ~/data/CIFAR-10/
6
+ img_size: 32
7
+ img_channels: 3
8
+ num_classes: 10
9
+ dataloader:
10
+ num_workers: 4
11
+ pin_memory: true
12
+ prefetch_factor: 2
13
+ model:
14
+ target: models.unet_categorial_adagn.UNetCategorialAdaGN
15
+ params:
16
+ in_channels: 3
17
+ out_channels: 3
18
+ dim: 128
19
+ dim_mults:
20
+ - 1
21
+ - 2
22
+ - 2
23
+ - 2
24
+ use_attn:
25
+ - false
26
+ - true
27
+ - true
28
+ - false
29
+ num_res_blocks: 2
30
+ num_classes: 10
31
+ attn_head_dims: 64
32
+ resblock_updown: true
33
+ dropout: 0.1
34
+ diffusion:
35
+ target: diffusions.cfg.ddpm_cfg.DDPMCFG
36
+ params:
37
+ total_steps: 1000
38
+ beta_schedule: cosine
39
+ beta_start: 0.0001
40
+ beta_end: 0.02
41
+ objective: pred_eps
42
+ var_type: fixed_large
43
+ train:
44
+ n_steps: 800000
45
+ batch_size: 128
46
+ micro_batch: 0
47
+ clip_grad_norm: 1.0
48
+ ema_decay: 0.9999
49
+ ema_gradual: true
50
+ print_freq: 400
51
+ save_freq: 10000
52
+ sample_freq: 5000
53
+ n_samples_each_class: 10
54
+ p_uncond: 0.2
55
+ optim:
56
+ target: torch.optim.AdamW
57
+ params:
58
+ lr: 0.0002
ddpm_cfg_cifar10/output-2023-05-06-19-03-22.log ADDED
The diff for this file is too large to render. See raw diff
 
ddpm_cfg_cifar10/samples.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2635b9b66a09850437399a314443ecbbd048ecc49976d4154de113eab996b630
3
+ size 33733915
ddpm_cfg_cifar10/tensorboard/events.out.tfevents.1683371004.ubuntu-WS-C621E-SAGE-Series.1587378.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:410910348663cc48e9828405190299c1dcf55c620234deab74ecea9ec457465f
3
+ size 78367060