Uploading NEXUS in asenella/MHD_NEXUS_2q2895rk
Browse files- README.md +13 -0
- decoders.pkl +3 -0
- encoders.pkl +3 -0
- environment.json +1 -0
- joint_encoder.pkl +3 -0
- model.pt +3 -0
- model_config.json +1 -0
- top_decoders.pkl +3 -0
- top_encoders.pkl +3 -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: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
|