search

EugenHotaj / pytorch-generative

Easy generative modeling in PyTorch
MIT License
428 stars 68 forks source link

VectorQuantizedVAE2 #30

Closed adammoss closed 2 years ago

adammoss commented 2 years ago

In the reproduce function of VQ-VAE-2, you use VectorQuantizedVAE rather than VectorQuantizedVAE2.

EugenHotaj commented 2 years ago

Good catch, thanks!