clementchadebec
commited on
Commit
•
bacc8b4
1
Parent(s):
2f84652
Uploading WAE_MMD in clementchadebec/reproduced_wae
Browse files- README.md +13 -0
- decoder.pkl +3 -0
- encoder.pkl +3 -0
- environment.json +1 -0
- model.pt +3 -0
- model_config.json +1 -0
README.md
ADDED
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
language: en
|
3 |
+
tags:
|
4 |
+
- pythae
|
5 |
+
license: apache-2.0
|
6 |
+
---
|
7 |
+
|
8 |
+
### Downloading this model from the Hub
|
9 |
+
This model was trained with pythae. It can be downloaded or reloaded using the method `load_from_hf_hub`
|
10 |
+
```python
|
11 |
+
>>> from pythae.models import AutoModel
|
12 |
+
>>> model = AutoModel.load_from_hf_hub(hf_hub_path="your_hf_username/repo_name")
|
13 |
+
```
|
decoder.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0a584864312a2649de2a511300f234f0d728f9d359bee3ef12ece1b765925acf
|
3 |
+
size 85921500
|
encoder.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d14d728c605a0685bdeae8981458a377733e3c57e1e1fe3844cdbbffddb242db
|
3 |
+
size 48312130
|
environment.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"name": "EnvironmentConfig", "python_version": "3.7"}
|
model.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:58c1933ea912832ea1255c8be96195a3cd5d29588d95885473e70a30fcff15bc
|
3 |
+
size 134223616
|
model_config.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
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": 2.0, "scales": [0.1, 0.2, 0.5, 1.0, 2.0, 5.0, 10.0]}
|