Hugging Face's logo

k-sert
/
gcDAE-checkpoints

Image-to-Image
PyTorch
diffusion
stable-diffusion
latent-space
representation-learning
hyperbolic-geometry
spherical-geometry
cub-200-2011
caltech-256
Model card Files
xet
 Community
k-sert commited on
Commit
d341710
·
verified ·
1 Parent(s): e96c535

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -20
README.md CHANGED
@@ -28,44 +28,44 @@ gcDAE studies how the geometry of different latent spaces affect diffusion-based
28
 
29
  The released checkpoints cover CUB-200-2011 and Caltech-256 with Euclidean, hyperbolic, and spherical Stage 2 latent spaces.
30
 
31
- ## Checkpoint Inventory
32
 
33
  Recommended repository layout:
34
 
35
  ```text
36
- gcDAE-checkpoints/
37
  ├── stage1/
38
  │ ├── cub200/
39
- │ │ └── epoch=000007.ckpt
40
  │ └── caltech256/
41
- │ └── epoch=000020.ckpt
42
  └── stage2/
43
  ├── cub200/
44
  │ ├── euc/
45
- │ │ └── epoch=000016.pt
46
  │ ├── hyp_c1/
47
- │ │ └── epoch=000032.pt
48
  │ └── sph_c1/
49
- │ └── epoch=000032.pt
50
  └── caltech256/
51
  ├── euc/
52
- │ └── epoch=000006.pt
53
  ├── hyp_c1/
54
- │ └── epoch=000012.pt
55
  └── sph_c1/
56
- └── epoch=000006.pt
57
  ```
58
 
59
- | Dataset | Stage | Geometry | Curvature | Suggested file |
60
- | :--- | :--- | :--- | :---: | :--- |
61
- | CUB-200-2011 | Stage 1 | diffusion backbone | - | `stage1/cub200/epoch=000007.ckpt` |
62
- | Caltech-256 | Stage 1 | diffusion backbone | - | `stage1/caltech256/epoch=000020.ckpt` |
63
- | CUB-200-2011 | Stage 2 | Euclidean | `c=0` | `stage2/cub200/euc/epoch=000016.ckpt` |
64
- | CUB-200-2011 | Stage 2 | Hyperbolic | `c=1` | `stage2/cub200/hyp_c1/epoch=000032.ckpt` |
65
- | CUB-200-2011 | Stage 2 | Spherical | `c=1` | `stage2/cub200/sph_c1/epoch=000032.ckpt` |
66
- | Caltech-256 | Stage 2 | Euclidean | `c=0` | `stage2/caltech256/euc/epoch=000006.ckpt` |
67
- | Caltech-256 | Stage 2 | Hyperbolic | `c=1` | `stage2/caltech256/hyp_c1/epoch=000012.ckpt` |
68
- | Caltech-256 | Stage 2 | Spherical | `c=1` | `stage2/caltech256/sph_c1/epoch=000006.ckpt` |
69
 
70
  ## Code
71
 
 
28
 
29
  The released checkpoints cover CUB-200-2011 and Caltech-256 with Euclidean, hyperbolic, and spherical Stage 2 latent spaces.
30
 
31
+ ## Released Checkpoints
32
 
33
  Recommended repository layout:
34
 
35
  ```text
36
+ gcDAE-models/
37
  ├── stage1/
38
  │ ├── cub200/
39
+ │ │ └── stage1_cub200.ckpt
40
  │ └── caltech256/
41
+ │ └── stage1_caltech256.ckpt
42
  └── stage2/
43
  ├── cub200/
44
  │ ├── euc/
45
+ │ │ └── stage2_cub200_euclidean.pt
46
  │ ├── hyp_c1/
47
+ │ │ └── stage2_cub200_hyperbolic_c1.pt
48
  │ └── sph_c1/
49
+ │ └── stage2_cub200_spherical_c1.pt
50
  └── caltech256/
51
  ├── euc/
52
+ │ └── stage2_caltech256_euclidean.pt
53
  ├── hyp_c1/
54
+ │ └── stage2_caltech256_hyperbolic_c1.pt
55
  └── sph_c1/
56
+ └── stage2_caltech256_spherical_c1.pt
57
  ```
58
 
59
+ | Dataset | Stage | Geometry | Curvature | Best Epoch |
60
+ | :--- | :--- | :--- | :---: | :---: |
61
+ | CUB-200-2011 | Stage 1 | Diffusion backbone | – | 7 |
62
+ | Caltech-256 | Stage 1 | Diffusion backbone | – | 20 |
63
+ | CUB-200-2011 | Stage 2 | Euclidean | `c=0` | 16 |
64
+ | CUB-200-2011 | Stage 2 | Hyperbolic | `c=1` | 32 |
65
+ | CUB-200-2011 | Stage 2 | Spherical | `c=1` | 32 |
66
+ | Caltech-256 | Stage 2 | Euclidean | `c=0` | 6 |
67
+ | Caltech-256 | Stage 2 | Hyperbolic | `c=1` | 12 |
68
+ | Caltech-256 | Stage 2 | Spherical | `c=1` | 6 |
69
 
70
  ## Code
71