xyfJASON commited on
Commit
d78b110
·
verified ·
1 Parent(s): 8c9b7d0

Add vqvae-celeba-64-512-norm

Browse files
vqvae-celeba-64-512-norm/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-norm/ckpt/step0499999/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6cba9d742255dec34e09b3c62ef1268e272045ea203102d0f9b6daec0ecf379a
3
+ size 19723985
vqvae-celeba-64-512-norm/ckpt/step0499999/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1236b117ec9fc3e86f598d14800f5980b93c59e1c5ec5c32a0e9cc708cb71b34
3
+ size 39452307
vqvae-celeba-64-512-norm/config-2024-07-29-06-42-22.yaml ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: true
28
+ train:
29
+ n_steps: 500000
30
+ batch_size: 256
31
+ coef_commit: 0.25
32
+ print_freq: 100
33
+ sample_freq: 10000
34
+ save_freq: 10000
35
+ optim:
36
+ target: torch.optim.Adam
37
+ params:
38
+ lr: 0.0004
vqvae-celeba-64-512-norm/output-2024-07-29-06-42-22.log ADDED
The diff for this file is too large to render. See raw diff
 
vqvae-celeba-64-512-norm/samples.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9903536c15e0bfdadc897ee0561ff1d5e490ec1ec20b423b543bed55eeb3ec38
3
+ size 23751815
vqvae-celeba-64-512-norm/tensorboard/events.out.tfevents.1722235342.jason-system.8394.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a1e6601987f7f34f47da2c6f0e7d58f78153f5ba16dd90c574fba03b9e266b8f
3
+ size 133917518