Uploading JNFGMC in asenella/translated_mmnist_resnets_2_JNFGMC_vz9gi42g
Browse files- README.md +13 -0
- decoders.pkl +3 -0
- encoders.pkl +3 -0
- environment.json +1 -0
- flows.pkl +3 -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:32456e9fc57d14f51ae1a28fbe40ebc2df6d3f6dd01ce6e36618fe924173b9a5
|
3 |
+
size 23753230
|
encoders.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:68ee199277c28b32f306d62e50a9b0d0b7b27c888356b28cc679e629868f6293
|
3 |
+
size 9517271
|
environment.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"name": "EnvironmentConfig", "python_version": "3.10"}
|
flows.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7a6670f6fbe22f0144c9a3e542b8191a295d619363632619116d47a6356a5f86
|
3 |
+
size 13396433
|
gmc_model.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a42f54be7627bf59e4b4ad40f0541006bd4b8592c08c440a2e790b392371f9d8
|
3 |
+
size 7203397
|
joint_encoder.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:55066e69e077aa6f89504f69a330dfdddc601ef7b5b9acb612f2284a2321ea5e
|
3 |
+
size 33542318
|
model.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fa123b22d72a767a7115ccaad571b461fa9687c2ac99fcbe408c825c0c632baf
|
3 |
+
size 87115562
|
model_config.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"name": "JNFGMCConfig", "n_modalities": 5, "latent_dim": 200, "input_dims": {"m0": [3, 28, 28], "m1": [3, 28, 28], "m2": [3, 28, 28], "m3": [3, 28, 28], "m4": [3, 28, 28]}, "uses_likelihood_rescaling": false, "rescale_factors": null, "decoders_dist": {"m0": "laplace", "m1": "laplace", "m2": "laplace", "m3": "laplace", "m4": "laplace"}, "decoder_dist_params": {"m0": {"scale": 0.75}, "m1": {"scale": 0.75}, "m2": {"scale": 0.75}, "m3": {"scale": 0.75}, "m4": {"scale": 0.75}}, "logits_to_std": "softplus", "custom_architectures": ["gmc_model", "encoders", "decoders", "joint_encoder", "flows"], "warmup": 150, "annealing": false, "alpha": 0.1, "nb_epochs_gmc": 80, "beta": 0.5, "gmc_config": {"name": "GMCConfig", "n_modalities": 5, "input_dims": {"m0": [3, 28, 28], "m1": [3, 28, 28], "m2": [3, 28, 28], "m3": [3, 28, 28], "m4": [3, 28, 28]}, "common_dim": 64, "latent_dim": 10, "temperature": 0.1, "custom_architectures": ["processors", "shared_encoder"], "loss": "between_modality_pairs"}, "add_reconstruction_terms": false, "divide_by_prior": true}
|