How to extract only the latent codes?

zalandoresearch / pytorch-vq-vae

PyTorch implementation of VQ-VAE by Aäron van den Oord et al.

MIT License

534 stars 101 forks source link

How to extract only the latent codes? #4

Open plaffa opened 5 years ago

plaffa commented 5 years ago

Hi! Thank you for the great upload. How exactly can I extract the latent code of an image? By that I mean the code of size i.e [1,8,8] and not [128,8,8]

Thanks!