-
Hi Jeremy, thank you for making your VQ-VAE code publicly available. As I was looking through `models/vqvae.py`, I had a question about the `_encoder(self, x)` function. In the function, you call `x =…
-
Hello,
There is a minor bug in /sonnet/python/modules/nets/vqvae.py
Please change keepdims=True to keep_dims=True
The underscore is missing and it fails for TF ver 1.4
Also it looks like the…
-
Nice implementation of the VQ-Straight through function!
However, when looking at the autograd graph there is an edge that is breaking the separated gradients for the reconstruction loss and the VQ…