Hugging Face's logo

xyfJASON
/
visual-tokenizer-pytorch

TensorBoard
Model card Files
xet
Metrics Community
xyfJASON commited on
Commit
453502f
·
verified ·
1 Parent(s): 347ee99

Add vqvae-celeba-64-512-entropy

Browse files
Files changed (7) hide show
  1. vqvae-celeba-64-512-entropy/ckpt/step0499999/meta.pt +3 -0
  2. vqvae-celeba-64-512-entropy/ckpt/step0499999/model.pt +3 -0
  3. vqvae-celeba-64-512-entropy/ckpt/step0499999/optimizer.pt +3 -0
  4. vqvae-celeba-64-512-entropy/config-2024-10-23-16-54-12.yaml +43 -0
  5. vqvae-celeba-64-512-entropy/output-2024-10-23-16-54-12.log +0 -0
  6. vqvae-celeba-64-512-entropy/samples.zip +3 -0
  7. vqvae-celeba-64-512-entropy/tensorboard/events.out.tfevents.1729673652.amax-ESC8000A-E11.2013035.0 +3 -0
vqvae-celeba-64-512-entropy/ckpt/step0499999/meta.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df99aa5f4769833bb82ff2e829751b4d07f62f0674714325cb3fc3ab673c2dc8
3
+ size 852
vqvae-celeba-64-512-entropy/ckpt/step0499999/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ede98b8e6cd35b288af52333c473b29b95de2e500d1bbccdaca756ef02200031
3
+ size 19723985
vqvae-celeba-64-512-entropy/ckpt/step0499999/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e3d78c79bc6d55d78cc9f80006de778301166faa21d3e1fc5f2ddbf9e7878f5
3
+ size 39452307
vqvae-celeba-64-512-entropy/config-2024-10-23-16-54-12.yaml ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ seed: 8888
2
+ data:
3
+ name: celeba
4
+ root: ~/data/CelebA/
5
+ img_size: 64
6
+ dataloader:
7
+ num_workers: 4
8
+ pin_memory: true
9
+ prefetch_factor: 2
10
+ encoder:
11
+ target: models.autoencoder.simple_cnn.Encoder
12
+ params:
13
+ img_channels: 3
14
+ hidden_dim: 256
15
+ n_resblocks: 2
16
+ decoder:
17
+ target: models.autoencoder.simple_cnn.Decoder
18
+ params:
19
+ img_channels: 3
20
+ hidden_dim: 256
21
+ n_resblocks: 2
22
+ quantizer:
23
+ target: models.quantizer.VectorQuantizer
24
+ params:
25
+ codebook_num: 512
26
+ codebook_dim: 64
27
+ l2_norm: false
28
+ use_ema_update: false
29
+ ema_decay: 0.99
30
+ use_entropy_reg: true
31
+ entropy_reg_temp: 0.01
32
+ train:
33
+ n_steps: 500000
34
+ batch_size: 256
35
+ coef_commit: 0.25
36
+ coef_entropy: 0.1
37
+ print_freq: 100
38
+ sample_freq: 10000
39
+ save_freq: 10000
40
+ optim:
41
+ target: torch.optim.Adam
42
+ params:
43
+ lr: 0.0004
vqvae-celeba-64-512-entropy/output-2024-10-23-16-54-12.log ADDED
The diff for this file is too large to render. See raw diff
 
vqvae-celeba-64-512-entropy/samples.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:814bce19df7a08ad36638239b49bee8c755d9d93595698498c36ff811a0d099c
3
+ size 23611034
vqvae-celeba-64-512-entropy/tensorboard/events.out.tfevents.1729673652.amax-ESC8000A-E11.2013035.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fe7574e4e8a4a5220a26200b15c96eb1597e881983a8a7fbd371946d9038b11
3
+ size 162901004