Add fsqvae-celeba-1000
Browse files- fsqvae-celeba-1000/ckpt/step0499999/meta.pt +3 -0
- fsqvae-celeba-1000/ckpt/step0499999/model.pt +3 -0
- fsqvae-celeba-1000/ckpt/step0499999/optimizer.pt +3 -0
- fsqvae-celeba-1000/config-2024-10-12-20-07-04.yaml +40 -0
- fsqvae-celeba-1000/output-2024-10-12-20-07-04.log +0 -0
- fsqvae-celeba-1000/samples.zip +3 -0
- fsqvae-celeba-1000/tensorboard/events.out.tfevents.1728734824.amax.2533317.0 +3 -0
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
|