Upload vae-celeba checkpoints and training logs
Browse files- vae-celeba/ckpt/step149999/meta.pt +3 -0
- vae-celeba/ckpt/step149999/model.pt +3 -0
- vae-celeba/ckpt/step149999/optimizer.pt +3 -0
- vae-celeba/config-2023-09-07-01-32-05.yaml +49 -0
- vae-celeba/output-2023-09-07-01-32-05.log +0 -0
- vae-celeba/samples.zip +3 -0
- vae-celeba/tensorboard/events.out.tfevents.1694021526.amax.1163181.0 +3 -0
vae-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-celeba/ckpt/step149999/model.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3708f4a2e6a295b05243787c4cf6ace856757ad7e0a1d336ac673c211fa1547e
|
3 |
+
size 44397300
|
vae-celeba/ckpt/step149999/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:731ea4f036ab73d7c2d26f2e78edb4fae92596981c93375671fafd5f481dbce3
|
3 |
+
size 88755653
|
vae-celeba/config-2023-09-07-01-32-05.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: 1.0
|
46 |
+
optim:
|
47 |
+
target: torch.optim.Adam
|
48 |
+
params:
|
49 |
+
lr: 0.02
|
vae-celeba/output-2023-09-07-01-32-05.log
ADDED
The diff for this file is too large to render.
See raw diff
|
|
vae-celeba/samples.zip
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:736207211f4f5482f8f606a03ed72000a426fc5e086d78ab35c8acf5af348452
|
3 |
+
size 57484318
|
vae-celeba/tensorboard/events.out.tfevents.1694021526.amax.1163181.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1abb91eab7089eaeb31033870e918f9abbaa2a1716b455ac33427474a9fcc484
|
3 |
+
size 23200546
|