asenella commited on
Commit
2b8f9e4
1 Parent(s): 5ed70f2

Uploading MVAE in asenella/mhd_config_1_MVAE_beta_5_scale_True_seed_1

Browse files
Files changed (6) hide show
  1. README.md +13 -0
  2. decoders.pkl +3 -0
  3. encoders.pkl +3 -0
  4. environment.json +1 -0
  5. model.pt +3 -0
  6. 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:b07c2f95c0adb691214217f1c6eff0df9fc17a9ca04f0bec986d14c2edf80b94
3
+ size 32018753
encoders.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31f4abbccf4fdf17c7b1d96edcfaaefa86108df77a42a20b50ce43223d9893a6
3
+ size 49378774
environment.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"name": "EnvironmentConfig", "python_version": "3.10"}
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0ac33619d53950d465ee367ac62ad8cdb9c2d2cc359aba688198ea79c44a0db
3
+ size 81330637
model_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"name": "MVAEConfig", "n_modalities": 3, "latent_dim": 64, "input_dims": {"image": [3, 28, 28], "audio": [1, 32, 128], "trajectory": [200]}, "uses_likelihood_rescaling": true, "rescale_factors": null, "decoders_dist": {"image": "normal", "audio": "normal", "trajectory": "normal"}, "decoder_dist_params": {}, "custom_architectures": ["encoders", "decoders"], "use_subsampling": true, "k": 1, "warmup": 100, "beta": 0.5}