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

Add fsqvae-celeba-512

Browse files
fsqvae-celeba-512/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-512/ckpt/step0499999/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b8e91ed3fcacc7f7f561ec1c3fbb670a35a0d990a4cc57214af5d068d7cad995
3
+ size 19467542
fsqvae-celeba-512/ckpt/step0499999/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03a9c4821dc869f25dfbfa091ba05239ee9a5306cf037e447bef39c5dc4e9f0e
3
+ size 38938966
fsqvae-celeba-512/config-2024-10-12-20-13-57.yaml ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ - 8
28
+ - 8
29
+ train:
30
+ n_steps: 500000
31
+ batch_size: 256
32
+ coef_commit: 0.25
33
+ print_freq: 100
34
+ sample_freq: 10000
35
+ save_freq: 10000
36
+ optim:
37
+ target: torch.optim.Adam
38
+ params:
39
+ lr: 0.0004
fsqvae-celeba-512/output-2024-10-12-20-13-57.log ADDED
The diff for this file is too large to render. See raw diff
 
fsqvae-celeba-512/samples.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5720f2e8ff55fb7cd14cdb1ab00c472865f27b1d9970ac4baed268ac83e0cb4
3
+ size 23667419
fsqvae-celeba-512/tensorboard/events.out.tfevents.1728735237.amax.2535268.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf48e156062320ff96ca2608d4fb491e741269b46f649ca07431673f843b7131
3
+ size 133917518