Uploading BetaTCVAE in clementchadebec/reproduced_beta_tc_vae

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:3642fd93323d87849d6d39aecf0962e134a4931b781c57156d94313184bfdd72
+size 31266122

encoder.pkl

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

environment.json

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

model.pt

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

model_config.json

@@ -0,0 +1 @@
+{"name": "BetaTCVAEConfig", "input_dim": [64, 64], "latent_dim": 10, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "bce", "alpha": 1.0, "beta": 6.0, "gamma": 1.0, "use_mss": false}