Added VAE results

xiaomn · Feb 14, 2020 · a277d3e · a277d3e
1 parent bb1f447
commit a277d3e
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -219,7 +219,7 @@ doesn't seem to work well.
 [28]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/recons_SWAE_49.png
 [29]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/MIWAE_29.png
 [30]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/recons_MIWAE_29.png
-[31]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/recons_VQVAE_29.png
+[31]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/recons_VQVAE_1.png
 
 [python-image]: https://img.shields.io/badge/Python-3.5-ff69b4.svg
 [python-url]: https://www.python.org/

diff --git a/assets/recons_VQVAE_1.png b/assets/recons_VQVAE_1.png
diff --git a/models/vq_vae.py b/models/vq_vae.py
@@ -7,7 +7,7 @@
 class VectorQuantizer(nn.Module):
     """
     Reference:
-    [1] https://github.com/zalandoresearch/pytorch-vq-vae
+    [1] https://github.com/deepmind/sonnet/blob/v2/sonnet/src/nets/vqvae.py
     """
     def __init__(self,
                  num_embeddings: int,