clementchadebec commited on
Commit
f0e2f40
1 Parent(s): 6eead9b

Uploading MVTCAE in clementchadebec/multivae_39

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:77c4ba5249a687f031cc5cf48b1a88c507d2781f0e73c282d8beec643ab01d1c
3
+ size 142269
encoders.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4bf8c74a4bd4b04d796d20893bf98e0f811ef49d5cd8c4d4f57e30da0f5b90d2
3
+ size 207226
environment.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"name": "EnvironmentConfig", "python_version": "3.9"}
model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13de2f27315c17a3c29f7d4b7788d583850beeda3108d33f0f0063a6659591a5
3
+ size 294421
model_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"name": "MVTCAEConfig", "n_modalities": 2, "latent_dim": 20, "input_dims": {"images": [1, 28, 28], "labels": [1, 10]}, "uses_likelihood_rescaling": true, "rescale_factors": {"images": 1, "labels": 50}, "decoders_dist": {"images": "normal", "labels": "categorical"}, "decoder_dist_params": {}, "custom_architectures": ["encoders", "decoders"], "alpha": 0.6666666666666666, "beta": 2.5}