Update README.md

ritheshkumar95 · May 22, 2018 · cb4702a · cb4702a
1 parent 43c9a71
commit cb4702a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -12,6 +12,15 @@ python vqvae.py --data-folder /tmp/miniimagenet --output-folder models/vqvae
 ```
 python pixelcnn_prior.py --data-folder /tmp/miniimagenet --model models/vqvae --output-folder models/pixelcnn_prior
 ```
+### Datasets Tested
+#### Image
+1. MNIST
+2. FashionMNIST
+3. CIFAR10
+4. Mini-ImageNet
+
+#### Video
+1. Atari 2600 - Boxing (OpenAI Gym) [code](https://github.com/ritheshkumar95/pytorch-vqvae/tree/evan/video)
 
 ### Reconstructions from VQ-VAE
 Top 4 rows are Original Images. Bottom 4 rows are Reconstructions.