Jonas Becker commited on
Commit
e8305d9
1 Parent(s): 7b7dac9

added binary models

Browse files
models/btcvae_celeba/data_samples.png ADDED
models/btcvae_celeba/model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61c1e59fc3e66c56bffefe2ce90e1da24a8969d2e0148a8a1db45544ed5d6d44
3
+ size 2020528
models/btcvae_celeba/onnx/decoder.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccd39ab6f904925a64f16bfe3ce94010a43810972a6c9ebb55d88c0283212818
3
+ size 1079224
models/btcvae_celeba/onnx/encoder.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5872ce5b75d088ce566f9bde58b23e5f8343b48671e91f7d0d025146b9443bb9
3
+ size 1119367
models/btcvae_celeba/posterior_traversals.gif ADDED

Git LFS Details

  • SHA256: d578da2bd9d09ac4e5971e71a82454c7dd8ff742dc06dbdc4534802714af72b6
  • Pointer size: 132 Bytes
  • Size of remote file: 6.13 MB
models/btcvae_celeba/posterior_traversals.png ADDED
models/btcvae_celeba/prior_traversals.png ADDED
models/btcvae_celeba/reconstruct.png ADDED
models/btcvae_celeba/reconstruct_traverse.png ADDED
models/btcvae_celeba/samples.png ADDED
models/btcvae_celeba/specs.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "batch_size": 64,
3
+ "betaB_G": 1000,
4
+ "betaB_finC": 25,
5
+ "betaB_initC": 0,
6
+ "betaH_B": 4,
7
+ "btcvae_A": 1,
8
+ "btcvae_B": 6.4,
9
+ "btcvae_G": 1,
10
+ "checkpoint_every": 100,
11
+ "dataset": "celeba",
12
+ "epochs": 200,
13
+ "eval_batchsize": 1000,
14
+ "experiment": "btcvae_celeba",
15
+ "factor_G": 6,
16
+ "img_size": [
17
+ 3,
18
+ 64,
19
+ 64
20
+ ],
21
+ "is_eval_only": false,
22
+ "is_metrics": false,
23
+ "latent_dim": 10,
24
+ "log_level": "info",
25
+ "loss": "btcvae",
26
+ "lr": 0.0005,
27
+ "lr_disc": 5e-05,
28
+ "model_type": "Burgess",
29
+ "name": "btcvae_celeba",
30
+ "no_cuda": false,
31
+ "no_progress_bar": true,
32
+ "no_test": false,
33
+ "rec_dist": "bernoulli",
34
+ "reg_anneal": 10000,
35
+ "seed": 1234
36
+ }
models/btcvae_celeba/test_losses.log ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dw_kl_loss": 0.003031336028000404,
3
+ "kl_loss": 0.10124790962106489,
4
+ "kl_loss_0": 0.009042695238085216,
5
+ "kl_loss_1": 0.011239859858169932,
6
+ "kl_loss_2": 0.007870119780742476,
7
+ "kl_loss_3": 0.011703663858874091,
8
+ "kl_loss_4": 0.008798975075406982,
9
+ "kl_loss_5": 0.010158935791165958,
10
+ "kl_loss_6": 0.009518803046841927,
11
+ "kl_loss_7": 0.008531809440387294,
12
+ "kl_loss_8": 0.015287692910931967,
13
+ "kl_loss_9": 0.009095354620459045,
14
+ "loss": 30.99244968057266,
15
+ "mi_loss": 0.3404961947737069,
16
+ "recon_loss": 32.045960975985224,
17
+ "tc_loss": -0.21828716259284558
18
+ }
models/btcvae_celeba/train_losses.log ADDED
The diff for this file is too large to render. See raw diff