xyfJASON commited on
Commit
bdf4046
·
verified ·
1 Parent(s): 7d4ac6b

Add fsqvae-celeba-1000

Browse files
fsqvae-celeba-1000/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
fsqvae-celeba-1000/ckpt/step0499999/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:effd101e05b713ccd9540d0d0bcca2c4f7904a79c026f6cdffab57e4be1d0c05
3
+ size 19469590
fsqvae-celeba-1000/ckpt/step0499999/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc548e78d8608dac4919a31cc44fa6086e70b3526405300eb737d852e663acce
3
+ size 38943062
fsqvae-celeba-1000/config-2024-10-12-20-07-04.yaml ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.FiniteScalarQuantizer
24
+ params:
25
+ levels:
26
+ - 8
27
+ - 5
28
+ - 5
29
+ - 5
30
+ train:
31
+ n_steps: 500000
32
+ batch_size: 256
33
+ coef_commit: 0.25
34
+ print_freq: 100
35
+ sample_freq: 10000
36
+ save_freq: 10000
37
+ optim:
38
+ target: torch.optim.Adam
39
+ params:
40
+ lr: 0.0004
fsqvae-celeba-1000/output-2024-10-12-20-07-04.log ADDED
The diff for this file is too large to render. See raw diff
 
fsqvae-celeba-1000/samples.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:486266ae3ccb9326321c733c051e2777b226294bccba0d894ddc75f1dd6c0507
3
+ size 23722182
fsqvae-celeba-1000/tensorboard/events.out.tfevents.1728734824.amax.2533317.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee932c3253044a42ba59683dc7256c11c50093f8ef0109285a84ba82f14f58a9
3
+ size 133917518