Uploading MMVAEPlus in asenella/MMVAEPlus_beta_5_scale_False_seed_2
Browse files- README.md +13 -0
- decoders.pkl +3 -0
- encoders.pkl +3 -0
- environment.json +1 -0
- model.pt +3 -0
- model_config.json +1 -0
README.md
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
tags:
|
| 4 |
+
- multivae
|
| 5 |
+
license: apache-2.0
|
| 6 |
+
---
|
| 7 |
+
|
| 8 |
+
### Downloading this model from the Hub
|
| 9 |
+
This model was trained with multivae. It can be downloaded or reloaded using the method `load_from_hf_hub`
|
| 10 |
+
```python
|
| 11 |
+
>>> from multivae.models import AutoModel
|
| 12 |
+
>>> model = AutoModel.load_from_hf_hub(hf_hub_path="your_hf_username/repo_name")
|
| 13 |
+
```
|
decoders.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b13d8e46e0d5e9a622db679b83f04a634fa99ac103934ddafd9db723810b958f
|
| 3 |
+
size 32018753
|
encoders.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3849a769b04fba228fa20412b9144d6582b36d9e995cbfcd35c9ecb0a1a27c7c
|
| 3 |
+
size 96902411
|
environment.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"name": "EnvironmentConfig", "python_version": "3.10"}
|
model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:faa4fcd3d20132bfb8f9c60b7ab8e5ae76da219d6f4824e46d0ca9e84fdcea6a
|
| 3 |
+
size 128845169
|
model_config.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"name": "MMVAEPlusConfig", "n_modalities": 3, "latent_dim": 32, "input_dims": {"image": [3, 28, 28], "audio": [1, 32, 128], "trajectory": [200]}, "uses_likelihood_rescaling": false, "rescale_factors": null, "decoders_dist": {"image": "normal", "audio": "normal", "trajectory": "normal"}, "decoder_dist_params": {}, "custom_architectures": ["encoders", "decoders"], "K": 1, "prior_and_posterior_dist": "laplace_with_softmax", "learn_shared_prior": false, "learn_modality_prior": true, "beta": 0.5, "modalities_specific_dim": 32, "reconstruction_option": "joint_prior"}
|