clementchadebec commited on
Commit
15c8d47
1 Parent(s): 9765a0c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -2
README.md CHANGED
@@ -2,12 +2,21 @@
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
  ```
 
 
 
 
 
 
 
 
 
 
2
  language: en
3
  tags:
4
  - pythae
5
+ - reproducibility
6
  license: apache-2.0
7
  ---
8
 
 
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="clementchadebec/reproduced_hvae")
13
  ```
14
+
15
+ ## Reproducibility
16
+ This trained model reproduces the results of the official implementation of [1].
17
+
18
+ | Model | Dataset | Metric | Obtained value | Reference value |
19
+ |:---:|:---:|:---:|:---:|:---:|
20
+ | BetaTCVAE | SPRITES | ELBO/Modified ELBO (after 50 epochs) | 710.41/85.54 | 712.26/86.40 |
21
+
22
+ [1] Ricky TQ Chen, Xuechen Li, Roger B Grosse, and David K Duvenaud. Isolating sources of disentanglement in variational autoencoders. Advances in neural information processing systems, 31, 2018.