asenella commited on
Commit
ecef50f
·
verified ·
1 Parent(s): 0f35d1f

Uploading JNFGMC in asenella/CUB_JNFGMC_v9wdsb19

Browse files
Files changed (9) hide show
  1. README.md +13 -0
  2. decoders.pkl +3 -0
  3. encoders.pkl +3 -0
  4. environment.json +1 -0
  5. flows.pkl +3 -0
  6. gmc_model.pkl +3 -0
  7. joint_encoder.pkl +3 -0
  8. model.pt +3 -0
  9. 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:914e9520f805070216939313f676e3c1ec014ba8bc11b55e617817d0616c58c9
3
+ size 26338276
encoders.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d19e4d579b55cc8514ed708fee21ff021722b6c8400103db7fb983f4c3212f8f
3
+ size 4898608
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:3a83e66c6feb3cbcc5691292d940081a24391735da13d9634752e9c2d1e0eb85
3
+ size 2851761
gmc_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:17f927d0bb6845156e3e2dce35c7061a20c0d7ab40e37117eae56f766eea694b
3
+ size 611004512
joint_encoder.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efdfe7c2335609d52d3ee6cf8407be6910e3030a016440f42ca8d86b8438dd1d
3
+ size 27977234
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4cd653cc2724c84982c65e9ea7c2dbd6796bf6006423c75ab1ef927ea6b8e328
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": 1.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}