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

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:a9285d0a1531ba9b9f88afebf35564ca087c368c46f17f3cdeba9641b0d352ea
+size 1369948

encoder.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e211ce1ecb69d54d4df8da1d222701b791185344db3e8483e9b63c04a568fb8a
+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:1814aa78c5bb20f29b9eb2153844f65437007c6c2f2722735d7f26f7ed733a60
+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}