Add fsqvae-celeba-512
Browse files- fsqvae-celeba-512/ckpt/step0499999/meta.pt +3 -0
- fsqvae-celeba-512/ckpt/step0499999/model.pt +3 -0
- fsqvae-celeba-512/ckpt/step0499999/optimizer.pt +3 -0
- fsqvae-celeba-512/config-2024-10-12-20-13-57.yaml +39 -0
- fsqvae-celeba-512/output-2024-10-12-20-13-57.log +0 -0
- fsqvae-celeba-512/samples.zip +3 -0
- fsqvae-celeba-512/tensorboard/events.out.tfevents.1728735237.amax.2535268.0 +3 -0
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
|