Commit
•
f0e2f40
1
Parent(s):
6eead9b
Uploading MVTCAE in clementchadebec/multivae_39
Browse files- README.md +13 -0
- decoders.pkl +3 -0
- encoders.pkl +3 -0
- environment.json +1 -0
- model.pt +3 -0
- 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}
|