search

dome272 / MaskGIT-pytorch

Pytorch implementation of MaskGIT: Masked Generative Image Transformer (https://arxiv.org/pdf/2202.04200.pdf)
MIT License
405 stars 35 forks source link

codebook loss #8

Open Dorniwang opened 2 years ago

Dorniwang commented 2 years ago

does the beta weighted wrong loss term of embedding loss which should be commitment loss in the vanilla va-vae?