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

Upload vae-beta20-celeba checkpoints and training logs

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