Uploading HVAE in clementchadebec/reproduced_hvae

decoder.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a0b07b1932669e5b4c37ba96e85daf77d78bc0db0b95cee8ad15932b11bd1dd9
-size 1970197
+oid sha256:0ae66d9e3b9b8d8432927b40310dd4fd81e24054560880050e1932567be05521
+size 1026429

encoder.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:04b9dd13effae96ca5ec5d42bb491425853933d7fcdd61548b812c8869f6a49d
-size 2031939
+oid sha256:63a3330a846bccd7a0aca2c0581f626cab743d2fac2b4bb5cd0e8678bb5ff35c
+size 1101296

model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d88b6990d454e63e67b6adc14caa0f20f23ddcd3cbc346b71c534979166a7d43
-size 3995067
+oid sha256:6204de1e9ae62e1b19060af5719e3a2b9efeae9f7cc05fca736f4ee0d6f6303e
+size 2121793

model_config.json

@@ -1 +1 @@
-{"name": "VAEConfig", "input_dim": [784], "latent_dim": 40, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "bce"}
+{"name": "HVAEConfig", "input_dim": [1, 28, 28], "latent_dim": 64, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "bce", "n_lf": 4, "eps_lf": 0.05, "beta_zero": 1.0, "learn_eps_lf": false, "learn_beta_zero": false}