Uploading JNFGMC in asenella/CUB_JNFGMC_03k2k4vm
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:cd17fb0d6312b07823aec40a7bc9c51d8d5f698c5184499fd19cf7490da2be5f
|
| 3 |
+
size 26338302
|
encoders.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5139913c6abc65c60d354ab08211186dc5556eb011c0f9494d60292583f04041
|
| 3 |
+
size 4898620
|
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:9fb9cb0ee55839a03198065d87ec475ad0bfa5350e275a3766209bcf52c41e45
|
| 3 |
+
size 2851805
|
gmc_model.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a49724341787d65bb5fa95aa924250b3a162706cda2cd77ecb71d27a447cf426
|
| 3 |
+
size 611004628
|
joint_encoder.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fd5602b0fccffb3630e41ec1aa3849b7c90e5711ef48a31d3625e1e2e8872651
|
| 3 |
+
size 27977258
|
model.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8e7e44cc395d65083239c1ad1cd4bce0b9d15becd88b60f5e6a17ec07e2a93d9
|
| 3 |
+
size 669435266
|
model_config.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"name": "JNFGMCConfig", "n_modalities": 2, "latent_dim": 64, "input_dims": null, "uses_likelihood_rescaling": true, "rescale_factors": {"image": 0.0026041666666666665, "text": 5.0}, "decoders_dist": {"image": "laplace", "text": "categorical"}, "decoder_dist_params": {"image": {"scale": 0.01}}, "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": 5.0, "gmc_config": {"name": "GMCConfig", "n_modalities": 2, "input_dims": null, "common_dim": 32, "latent_dim": 32, "temperature": 0.1, "custom_architectures": ["processors", "shared_encoder"], "loss": "between_modality_pairs"}, "add_reconstruction_terms": false, "divide_by_prior": true}
|