asenella commited on
Commit
abc02a4
·
verified ·
1 Parent(s): 7ff866d

Uploading NEXUS in asenella/MHD_NEXUS_2q2895rk

Browse files
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:08dd7d2a9036ac210ad97fc5b65c7a225499b31f8357ff5b38f6a33d9d7f446d
3
+ size 32445154
encoders.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9fce905548f6017b2de214469a9f907d95a8e6d7ddc0f2a66104e6f5b3c03c8a
3
+ size 50231321
environment.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"name": "EnvironmentConfig", "python_version": "3.10"}
joint_encoder.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d1ce894a0077ed3f662bd19cde27c86b8bf275b4f0dfce23a3a0eda4318f204
3
+ size 2397794
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d25087ed71ab4af7d5b5848e63c4132a8cddea06194f8c0a1359af8395dffd7d
3
+ size 92652550
model_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"name": "NexusConfig", "n_modalities": 3, "latent_dim": 64, "input_dims": {"image": [3, 28, 28], "audio": [1, 32, 128], "trajectory": [200]}, "uses_likelihood_rescaling": true, "rescale_factors": {"image": 1.0, "trajectory": 50.0, "audio": 1.0}, "decoders_dist": {"image": "normal", "audio": "normal", "trajectory": "normal"}, "decoder_dist_params": {}, "custom_architectures": ["encoders", "decoders", "top_encoders", "top_decoders", "joint_encoder"], "modalities_specific_dim": {"image": 64, "audio": 128, "trajectory": 16}, "bottom_betas": {"image": 1.0, "trajectory": 1.0, "audio": 1.0}, "dropout_rate": 0.2, "msg_dim": 512, "aggregator": "mean", "top_beta": 1.0, "gammas": {"image": 1.0, "trajectory": 50.0, "audio": 1.0}, "warmup": 20, "adapt_top_decoder_variance": null}
top_decoders.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e46fba695c7b9d3e0ef2afedeb2b7a20c0ce117aa23138a80a751673b87c5f59
3
+ size 7249631
top_encoders.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37fafb66e4f5bdb0f81434b1f265cdc4ff140401a95cb6e2e5914a3268ac55b4
3
+ size 443202