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.

Sign up for GitHub

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

Jump to bottom

What is the image input for inference? #435

Open

SenHe opened this issue Jun 29, 2022 · 1 comment

SenHe commented Jun 29, 2022 •

edited

Loading

Thanks for this great work!

After going through the code, I got some questions.

In the first stage of training discrete VAE, we already trained a code book. Why we don't use it for second stage training but initialize a new code book for images.
During training, we use the original image as input. During inference, how to set the image input? Is it a random noise with size 3x256x256? How do we do the casual attention in transformer for inference?

kingnobro commented Sep 4, 2022

After reading codes, I also don't know why there is another new code book. Do you have any idea now?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment