Upload vae-beta20-celeba checkpoints and training logs
Browse files- vae-beta20-celeba/ckpt/step149999/meta.pt +3 -0
- vae-beta20-celeba/ckpt/step149999/model.pt +3 -0
- vae-beta20-celeba/ckpt/step149999/optimizer.pt +3 -0
- vae-beta20-celeba/config-2023-09-08-17-11-55.yaml +49 -0
- vae-beta20-celeba/output-2023-09-08-17-11-55.log +0 -0
- vae-beta20-celeba/samples.zip +3 -0
- vae-beta20-celeba/tensorboard/events.out.tfevents.1694164316.amax.1396664.0 +3 -0
vae-beta20-celeba/ckpt/step149999/meta.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d73f91e6f9f0e8222c08d532226feb041d885e7aed885e0536b91a6887793879
|
| 3 |
+
size 425
|
vae-beta20-celeba/ckpt/step149999/model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f02646218bc4f15dcc13e99b2e86811a472152b6d57a6a01ebb092fd254333c4
|
| 3 |
+
size 44397300
|
vae-beta20-celeba/ckpt/step149999/optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3268b7c43ffcb882fd4c69f3d84b72e9e2ca4b088b9804cdde1ba97e6a948363
|
| 3 |
+
size 88755653
|
vae-beta20-celeba/config-2023-09-08-17-11-55.yaml
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
seed: 8888
|
| 2 |
+
data:
|
| 3 |
+
target: datasets.celeba.CelebA
|
| 4 |
+
params:
|
| 5 |
+
root: ~/data/CelebA/
|
| 6 |
+
img_size: 64
|
| 7 |
+
split: train
|
| 8 |
+
dataloader:
|
| 9 |
+
num_workers: 4
|
| 10 |
+
pin_memory: true
|
| 11 |
+
prefetch_factor: 2
|
| 12 |
+
encoder:
|
| 13 |
+
target: models.simple_cnn.Encoder
|
| 14 |
+
params:
|
| 15 |
+
z_dim: 128
|
| 16 |
+
in_dim: 3
|
| 17 |
+
dim: 64
|
| 18 |
+
dim_mults:
|
| 19 |
+
- 1
|
| 20 |
+
- 2
|
| 21 |
+
- 4
|
| 22 |
+
- 8
|
| 23 |
+
with_bn: true
|
| 24 |
+
decoder:
|
| 25 |
+
target: models.simple_cnn.Decoder
|
| 26 |
+
params:
|
| 27 |
+
z_dim: 128
|
| 28 |
+
dim: 64
|
| 29 |
+
dim_mults:
|
| 30 |
+
- 8
|
| 31 |
+
- 4
|
| 32 |
+
- 2
|
| 33 |
+
- 1
|
| 34 |
+
out_dim: 3
|
| 35 |
+
with_bn: true
|
| 36 |
+
with_tanh: true
|
| 37 |
+
train:
|
| 38 |
+
n_steps: 150000
|
| 39 |
+
batch_size: 256
|
| 40 |
+
resume: null
|
| 41 |
+
print_freq: 100
|
| 42 |
+
sample_freq: 1000
|
| 43 |
+
save_freq: 10000
|
| 44 |
+
n_samples: 64
|
| 45 |
+
coef_kl: 20
|
| 46 |
+
optim:
|
| 47 |
+
target: torch.optim.Adam
|
| 48 |
+
params:
|
| 49 |
+
lr: 0.02
|
vae-beta20-celeba/output-2023-09-08-17-11-55.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
vae-beta20-celeba/samples.zip
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c13e219bb2e8d46c1b66637f9710737ffc5b3c42a1cc3556755096fa5d955f31
|
| 3 |
+
size 52878805
|
vae-beta20-celeba/tensorboard/events.out.tfevents.1694164316.amax.1396664.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:06e7a89d72865c5b4cbfd896ce8d82e62ad4197428ef9e1aa71e8f5f9a03ebef
|
| 3 |
+
size 23200546
|