Uploading PoincareVAE in clementchadebec/reproduced_wrapped_poincare_vae

README.md

@@ -0,0 +1,13 @@
+---
+language: en
+tags:
+- pythae
+license: apache-2.0
+---
+
+### Downloading this model from the Hub
+This model was trained with pythae. It can be downloaded or reloaded using the method `load_from_hf_hub`
+```python
+>>> from pythae.models import AutoModel
+>>> model = AutoModel.load_from_hf_hub(hf_hub_path="your_hf_username/repo_name")
+```

decoder.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5268898aee57f2293ee816ff1b0db8c96d1a6d689932fa19ddb52b10c9d2601e
+size 1917596

encoder.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:63869ad9a707722a2be018d84d4ad7fb099cc061f148eb26017b9fa5e26517d9
+size 1915139

environment.json

@@ -0,0 +1 @@
+{"name": "EnvironmentConfig", "python_version": "3.8"}

model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b59f80f03e50138c692ed095a83e212b5bccff64dda54966f70b272494312fad
+size 3825217

model_config.json

@@ -0,0 +1 @@
+{"name": "PoincareVAEConfig", "input_dim": [1, 28, 28], "latent_dim": 10, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "bce", "prior_distribution": "wrapped_normal", "posterior_distribution": "wrapped_normal", "curvature": 0.7}