Uploading VAE in onlytojay/resnet_vae_celeba

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:7f26cc69eb364f55afe8faef3d89f7082aef9b1317d0e6b71674df78410aa91b
+size 3940325

encoder.pkl

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

environment.json

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

model.pt

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

model_config.json

@@ -0,0 +1 @@
+{"name": "VAEConfig", "input_dim": [3, 64, 64], "latent_dim": 64, "uses_default_encoder": false, "uses_default_decoder": false, "reconstruction_loss": "mse"}