Uploading DisentangledBetaVAE in Allayte/Disentangled-BETA-VAE-TMNIST

README.md

@@ -0,0 +1,13 @@
+---
+language: en
+tags:
+- pythae
+license: apache-2.0
+---
+
+### Downloading this model from the Hub
+This model was trained with pythae. It can be downloaded or reloaded using the method `load_from_hf_hub`
+```python
+>>> from pythae.models import AutoModel
+>>> model = AutoModel.load_from_hf_hub(hf_hub_path="your_hf_username/repo_name")
+```

decoder.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b6fe35e895381512c02db5b547f752137cdeb645595e51101e65f9fdc3cb8ec
+size 1369948

encoder.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d713061c317254f9fd814fdd9e3584da68dc5fdc6b85f637641310b4b801f59
+size 1725188

environment.json

@@ -0,0 +1 @@
+{"name": "EnvironmentConfig", "python_version": "3.8"}

model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8dd832785bc6eba9e86453db06eb2cb6eabb637a8b9f0842f01f20b19f788dff
+size 3077047

model_config.json

@@ -0,0 +1 @@
+{"name": "DisentangledBetaVAEConfig", "input_dim": [1, 28, 28], "latent_dim": 16, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "mse", "beta": 5.0, "C": 30.0, "warmup_epoch": 25}