search

rosinality / vq-vae-2-pytorch

Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch
Other
1.62k stars 272 forks source link

How to condition the sampler #73

Open parthnatekar opened 1 year ago

parthnatekar commented 1 year ago

Hi, I've trained the VQVAE and the Pixel snail models on a dataset with multiple classes.

However when I sample, I always seem to be getting different images of the same class.

How can I condition the sampler to give different classes?