Uploading WAE_MMD in onlytojay/resnet_wae_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:de38b8f512531821cb8722d5efeb7ae0d1b6c84caeccbb9588f29b0bb5b86441
+size 3940325

encoder.pkl

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

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:c8449a51166c4a5fc6cc18dfbe4233d54a8f7165e7d595df7a27a8424b1e4031
+size 6506147

model_config.json

@@ -0,0 +1 @@
+{"name": "WAE_MMD_Config", "input_dim": [3, 64, 64], "latent_dim": 64, "uses_default_encoder": false, "uses_default_decoder": false, "kernel_choice": "imq", "reg_weight": 100.0, "kernel_bandwidth": 1.0, "scales": [0.1, 0.2, 0.5, 1.0, 2.0, 5.0, 10.0], "reconstruction_loss_scale": 1.0}