xyfJASON commited on
Commit
a719ed7
·
1 Parent(s): 78e8601

Upload vae-celeba checkpoints and training logs

Browse files
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