Hugging Face's logo

UT-Austin-PML
/
BetaDiffusion

Model card Files Community
mingyuanzhou commited on
Commit
bccc6c4
•
1 Parent(s): fbde75e

add checkpoints for plotting

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. checkpoint_Beta_ELBO_B288/.DS_Store +0 -0
  2. checkpoint_Beta_ELBO_B288/.ipynb_checkpoints/betadiff_x0_hat-checkpoint.png +0 -0
  3. checkpoint_Beta_ELBO_B288/.ipynb_checkpoints/image90-checkpoint.png +0 -0
  4. checkpoint_Beta_ELBO_B288/betadiff_x0_hat.png +0 -0
  5. checkpoint_Beta_ELBO_B288/checkpoint.txt +3 -0
  6. checkpoint_Beta_ELBO_B288/image27.png +0 -0
  7. checkpoint_Beta_ELBO_B288/image27_1.png +0 -0
  8. checkpoint_Beta_ELBO_B288/image29.png +0 -0
  9. checkpoint_Beta_ELBO_B288/image29_1.png +0 -0
  10. checkpoint_Beta_ELBO_B288/image30.png +0 -0
  11. checkpoint_Beta_ELBO_B288/image30_1.png +0 -0
  12. checkpoint_Beta_ELBO_B288/image54.png +0 -0
  13. checkpoint_Beta_ELBO_B288/image54_1.png +0 -0
  14. checkpoint_Beta_ELBO_B288/image74.png +0 -0
  15. checkpoint_Beta_ELBO_B288/image74_1.png +0 -0
  16. checkpoint_Beta_ELBO_B288/image80.png +0 -0
  17. checkpoint_Beta_ELBO_B288/image80_1.png +0 -0
  18. checkpoint_Beta_ELBO_B288/image9.png +0 -0
  19. checkpoint_Beta_ELBO_B288/image90.png +0 -0
  20. checkpoint_Beta_ELBO_B288/image90_1.png +0 -0
  21. checkpoint_Beta_ELBO_B288/image93.png +0 -0
  22. checkpoint_Beta_ELBO_B288/image93_1.png +0 -0
  23. checkpoint_Beta_ELBO_B288/image9_1.png +0 -0
  24. checkpoint_Beta_ELBO_B288/network-snapshot-200000.pkl +3 -0
  25. checkpoint_Beta_ELBO_B288/training_options.json +77 -0
  26. checkpoint_Beta_KLUB_B288/.DS_Store +0 -0
  27. checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/betadiff_x0_hat-checkpoint.png +0 -0
  28. checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image27-checkpoint.png +0 -0
  29. checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image74-checkpoint.png +0 -0
  30. checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image9-checkpoint.png +0 -0
  31. checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image90-checkpoint.png +0 -0
  32. checkpoint_Beta_KLUB_B288/betadiff_x0_hat.png +0 -0
  33. checkpoint_Beta_KLUB_B288/image27.png +0 -0
  34. checkpoint_Beta_KLUB_B288/image27_1.png +0 -0
  35. checkpoint_Beta_KLUB_B288/image29.png +0 -0
  36. checkpoint_Beta_KLUB_B288/image29_1.png +0 -0
  37. checkpoint_Beta_KLUB_B288/image30.png +0 -0
  38. checkpoint_Beta_KLUB_B288/image30_1.png +0 -0
  39. checkpoint_Beta_KLUB_B288/image54.png +0 -0
  40. checkpoint_Beta_KLUB_B288/image54_1.png +0 -0
  41. checkpoint_Beta_KLUB_B288/image74.png +0 -0
  42. checkpoint_Beta_KLUB_B288/image74_1.png +0 -0
  43. checkpoint_Beta_KLUB_B288/image80.png +0 -0
  44. checkpoint_Beta_KLUB_B288/image80_1.png +0 -0
  45. checkpoint_Beta_KLUB_B288/image9.png +0 -0
  46. checkpoint_Beta_KLUB_B288/image90.png +0 -0
  47. checkpoint_Beta_KLUB_B288/image90_1.png +0 -0
  48. checkpoint_Beta_KLUB_B288/image93.png +0 -0
  49. checkpoint_Beta_KLUB_B288/image93_1.png +0 -0
  50. checkpoint_Beta_KLUB_B288/image9_1.png +0 -0
checkpoint_Beta_ELBO_B288/.DS_Store ADDED
Binary file (6.15 kB). View file
 
checkpoint_Beta_ELBO_B288/.ipynb_checkpoints/betadiff_x0_hat-checkpoint.png ADDED
checkpoint_Beta_ELBO_B288/.ipynb_checkpoints/image90-checkpoint.png ADDED
checkpoint_Beta_ELBO_B288/betadiff_x0_hat.png ADDED
checkpoint_Beta_ELBO_B288/checkpoint.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ Link to checkpoint:
2
+
3
+ https://drive.google.com/file/d/1jaMaDl-R1IVFlPlrj9nd9dNZWIutUU1s/view?usp=drive_link
checkpoint_Beta_ELBO_B288/image27.png ADDED
checkpoint_Beta_ELBO_B288/image27_1.png ADDED
checkpoint_Beta_ELBO_B288/image29.png ADDED
checkpoint_Beta_ELBO_B288/image29_1.png ADDED
checkpoint_Beta_ELBO_B288/image30.png ADDED
checkpoint_Beta_ELBO_B288/image30_1.png ADDED
checkpoint_Beta_ELBO_B288/image54.png ADDED
checkpoint_Beta_ELBO_B288/image54_1.png ADDED
checkpoint_Beta_ELBO_B288/image74.png ADDED
checkpoint_Beta_ELBO_B288/image74_1.png ADDED
checkpoint_Beta_ELBO_B288/image80.png ADDED
checkpoint_Beta_ELBO_B288/image80_1.png ADDED
checkpoint_Beta_ELBO_B288/image9.png ADDED
checkpoint_Beta_ELBO_B288/image90.png ADDED
checkpoint_Beta_ELBO_B288/image90_1.png ADDED
checkpoint_Beta_ELBO_B288/image93.png ADDED
checkpoint_Beta_ELBO_B288/image93_1.png ADDED
checkpoint_Beta_ELBO_B288/image9_1.png ADDED
checkpoint_Beta_ELBO_B288/network-snapshot-200000.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f85ac2428caa93be7bd7ef7e280a658a1379cd0869f2bd36614f2cbca2578d0e
3
+ size 223199962
checkpoint_Beta_ELBO_B288/training_options.json ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset_kwargs": {
3
+ "class_name": "training.dataset.ImageFolderDataset",
4
+ "path": "datasets/cifar10-32x32.zip",
5
+ "use_labels": false,
6
+ "xflip": false,
7
+ "cache": true,
8
+ "resolution": 32,
9
+ "max_size": 50000
10
+ },
11
+ "data_loader_kwargs": {
12
+ "pin_memory": true,
13
+ "num_workers": 1,
14
+ "prefetch_factor": 2
15
+ },
16
+ "network_kwargs": {
17
+ "model_type": "SongUNet",
18
+ "embedding_type": "positional",
19
+ "encoder_type": "standard",
20
+ "decoder_type": "standard",
21
+ "channel_mult_noise": 1,
22
+ "resample_filter": [
23
+ 1,
24
+ 1
25
+ ],
26
+ "model_channels": 128,
27
+ "channel_mult": [
28
+ 2,
29
+ 2,
30
+ 2
31
+ ],
32
+ "class_name": "training.networks.BetaDiffPrecond",
33
+ "augment_dim": 9,
34
+ "dropout": 0.13,
35
+ "use_fp16": false
36
+ },
37
+ "loss_kwargs": {
38
+ "class_name": "training.loss.BetaDiffLoss",
39
+ "eta": 10000.0,
40
+ "sigmoid_start": 10.0,
41
+ "sigmoid_end": -13.0,
42
+ "sigmoid_power": 1.0,
43
+ "Scale": 0.39,
44
+ "Shift": 0.6,
45
+ "lossType": "KLUB_AS"
46
+ },
47
+ "optimizer_kwargs": {
48
+ "class_name": "torch.optim.Adam",
49
+ "lr": 0.0002,
50
+ "betas": [
51
+ 0.9,
52
+ 0.999
53
+ ],
54
+ "eps": 1e-08
55
+ },
56
+ "augment_kwargs": {
57
+ "class_name": "training.augment.AugmentPipe",
58
+ "p": 0.12,
59
+ "xflip": 100000000.0,
60
+ "yflip": 1,
61
+ "scale": 1,
62
+ "rotate_frac": 1,
63
+ "aniso": 1,
64
+ "translate_frac": 1
65
+ },
66
+ "total_kimg": 200000,
67
+ "ema_halflife_kimg": 500,
68
+ "batch_size": 288,
69
+ "batch_gpu": null,
70
+ "loss_scaling": 1.0,
71
+ "cudnn_benchmark": true,
72
+ "kimg_per_tick": 50,
73
+ "snapshot_ticks": 50,
74
+ "state_dump_ticks": 500,
75
+ "seed": 17439077,
76
+ "run_dir": "betadiff-train-runs/00018-cifar10-32x32-uncond-ddpmpp-betadiff-gpus8-batch288-fp32"
77
+ }
checkpoint_Beta_KLUB_B288/.DS_Store ADDED
Binary file (6.15 kB). View file
 
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/betadiff_x0_hat-checkpoint.png ADDED
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image27-checkpoint.png ADDED
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image74-checkpoint.png ADDED
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image9-checkpoint.png ADDED
checkpoint_Beta_KLUB_B288/.ipynb_checkpoints/image90-checkpoint.png ADDED
checkpoint_Beta_KLUB_B288/betadiff_x0_hat.png ADDED
checkpoint_Beta_KLUB_B288/image27.png ADDED
checkpoint_Beta_KLUB_B288/image27_1.png ADDED
checkpoint_Beta_KLUB_B288/image29.png ADDED
checkpoint_Beta_KLUB_B288/image29_1.png ADDED
checkpoint_Beta_KLUB_B288/image30.png ADDED
checkpoint_Beta_KLUB_B288/image30_1.png ADDED
checkpoint_Beta_KLUB_B288/image54.png ADDED
checkpoint_Beta_KLUB_B288/image54_1.png ADDED
checkpoint_Beta_KLUB_B288/image74.png ADDED
checkpoint_Beta_KLUB_B288/image74_1.png ADDED
checkpoint_Beta_KLUB_B288/image80.png ADDED
checkpoint_Beta_KLUB_B288/image80_1.png ADDED
checkpoint_Beta_KLUB_B288/image9.png ADDED
checkpoint_Beta_KLUB_B288/image90.png ADDED
checkpoint_Beta_KLUB_B288/image90_1.png ADDED
checkpoint_Beta_KLUB_B288/image93.png ADDED
checkpoint_Beta_KLUB_B288/image93_1.png ADDED
checkpoint_Beta_KLUB_B288/image9_1.png ADDED