Uploading MVAE in asenella/reproduce_mvae_mnist

README.md

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

decoders.pkl

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

encoders.pkl

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

model.pt

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

model_config.json

@@ -0,0 +1 @@
+{"name": "MVAEConfig", "n_modalities": 2, "latent_dim": 64, "input_dims": {"images": [1, 28, 28], "labels": [1]}, "uses_likelihood_rescaling": true, "rescale_factors": {"images": 1, "labels": 50}, "decoders_dist": {"images": "bernoulli", "labels": "categorical"}, "decoder_dist_params": {}, "custom_architectures": ["encoders", "decoders"], "k": 0, "warmup": 200}