Update README.md

igul222 · Jan 17, 2017 · 4b0c4a9 · 4b0c4a9
1 parent 6ebaede
commit 4b0c4a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -2,7 +2,7 @@
 
 Code for the models in [PixelVAE: A Latent Variable Model for Natural Images](https://arxiv.org/abs/1611.05013)
 
-## MNIST
+## Binarized MNIST
 
 To train:
 
@@ -16,7 +16,7 @@ To evaluate, take the weights of the model with best validation score from the a
 python models/mnist_pixelvae_evaluate.py -L 12 -fs 5 -algo cond_z_bias -dpx 16 -ldim 16 -w path/to/weights.pkl
 ```
 
-## Other datasets
+## Real-valued MNIST, LSUN Bedrooms, 64x64 ImageNet, and other datasets
 
 To train, evaluate, and generate samples: