-
Notifications
You must be signed in to change notification settings - Fork 154
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Issue with Changing embedding_dim
in VQ-VAE Model
#138
Comments
Hi @arsh-rl, Actually, this value is set automatically to either the number of channels of your encoded sample or the size of your latent space in case of flattened encoded input. This is needed to be able to quantized the encoded sample within the codebook. To change this value you should either adapt you encoder architecture to output a sample with the required embedded dimension if you created you own or change the I hope this helps. Best, Clément |
Hi @clementchadebec.
Thank you for creating this repository.
I am attempting to train a VQ-VAE model, but I couldn't find an
embedding_dim
argument in either theVQVAEconfig
orVQVAE
classes to assign a value to it.From what I have found, the only place where the
embedding_dim
is assigned is inside the_set_quantizer
function of theVQVAE
class, which is hard-coded to one.z.shape[-1]
always holds the value 1.The text was updated successfully, but these errors were encountered: