Uploading JNFGMC in asenella/MHD_JNFGMC_rzm5fs1a
Browse files- README.md +13 -0
- decoders.pkl +3 -0
- environment.json +1 -0
- gmc_model.pkl +3 -0
- joint_encoder.pkl +3 -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:42a6443cff33b8833bf89883df0aaf069cc020e62159388c936dea482519a941
|
3 |
+
size 32019163
|
environment.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"name": "EnvironmentConfig", "python_version": "3.10"}
|
gmc_model.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c2ed5e190406390f6a065fa051c64045869d36c273a71f3c85c7092c37f807f6
|
3 |
+
size 48028896
|
joint_encoder.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c23e5fdb7978cb2961a83d7741e1e62c6af412ae4d568b2e434e43558b38a267
|
3 |
+
size 51089602
|
model.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:92a22670d156158063d573d351468b2654394f2946199c47cb53c41f081e2579
|
3 |
+
size 137868558
|
model_config.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"name": "JNFGMCConfig", "n_modalities": 3, "latent_dim": 64, "input_dims": {"image": [3, 28, 28], "audio": [1, 32, 128], "trajectory": [200]}, "uses_likelihood_rescaling": true, "rescale_factors": null, "decoders_dist": {"image": "normal", "audio": "normal", "trajectory": "normal"}, "decoder_dist_params": {}, "custom_architectures": ["gmc_model", "decoders", "joint_encoder"], "warmup": 200, "annealing": false, "alpha": 0.1, "nb_epochs_gmc": 100, "beta": 2.5, "gmc_config": {"name": "GMCConfig", "n_modalities": 3, "input_dims": {"image": [3, 28, 28], "audio": [1, 32, 128], "trajectory": [200]}, "common_dim": 16, "latent_dim": 16, "temperature": 0.2, "custom_architectures": ["processors", "shared_encoder"], "loss": "between_modality_pairs"}, "logits_to_std": "standard"}
|