arsh-rl

Issue with Changing `embedding_dim` in VQ-VAE Model

1

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 the `VQVAEconfig` or `VQVAE` classes...