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

Is this implementation for the MoE correct? #2

Open

daydreamer2023 opened this issue May 3, 2021 · 0 comments

daydreamer2023 commented May 3, 2021

QMVAE-mmdgm/mnist_svhn/model.py

Lines 261 to 262 in 8bec59c

    
           mu, logvar = torch.cat([mu[1,:B//2], mu[2, B//2:]]), \ 
        
                        torch.cat([logvar[1,:B//2], logvar[2, B//2:]])

Hi!
Is this implementation for the MoE correct?
It lead to the result:

recon_2x0_004

recon_2x1_004

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