Hugging Face's logo

xyfJASON
/
Diffusion-Models-Implementations

TensorBoard
Model card Files Metrics Community
xyfJASON commited on
Commit
e332f9b
•
1 Parent(s): d06e700

Upload classifier_free_cifar10 checkpoints and training logs

Browse files
Files changed (8) hide show
  1. classifier_free_cifar10/ckpt/step799999/ema.pt +3 -0
  2. classifier_free_cifar10/ckpt/step799999/meta.pt +3 -0
  3. classifier_free_cifar10/ckpt/step799999/model.pt +3 -0
  4. classifier_free_cifar10/ckpt/step799999/optimizer.pt +3 -0
  5. classifier_free_cifar10/config-2023-05-06-19-03-22.yaml +58 -0
  6. classifier_free_cifar10/output-2023-05-06-19-03-22.log +0 -0
  7. classifier_free_cifar10/samples.zip +3 -0
  8. classifier_free_cifar10/tensorboard/events.out.tfevents.1683371004.ubuntu-WS-C621E-SAGE-Series.1587378.0 +3 -0
classifier_free_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
classifier_free_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
classifier_free_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
classifier_free_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
classifier_free_cifar10/config-2023-05-06-19-03-22.yaml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ seed: 2022
2
+ data:
3
+ name: CIFAR-10
4
+ dataroot: /data/xuyifeng/data/CIFAR-10/
5
+ img_size: 32
6
+ img_channels: 3
7
+ num_classes: 10
8
+ dataloader:
9
+ num_workers: 4
10
+ pin_memory: true
11
+ prefetch_factor: 2
12
+ micro_batch: 0
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
+ ema_decay: 0.9999
35
+ ema_gradual: true
36
+ diffusion:
37
+ target: diffusions.guided_free.GuidedFree
38
+ params:
39
+ total_steps: 1000
40
+ beta_schedule: cosine
41
+ beta_start: 0.0001
42
+ beta_end: 0.02
43
+ objective: pred_eps
44
+ var_type: fixed_large
45
+ train:
46
+ n_steps: 800000
47
+ batch_size: 128
48
+ clip_grad_norm: 1.0
49
+ resume: null
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
classifier_free_cifar10/output-2023-05-06-19-03-22.log ADDED
The diff for this file is too large to render. See raw diff
 
classifier_free_cifar10/samples.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3c10b7fb7f3e9b53a52e9aebd8bc8b606f2978f14f44bf5c82e761badec25e2
3
+ size 33733915
classifier_free_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