enhancing-transformers icon indicating copy to clipboard operation
enhancing-transformers copied to clipboard

Published 20 hours ago • verified publisher icon thuanz123

Incomplete implementation of RQ-VAE

Open JJJYmmm opened this issue 10 months ago • 0 comments

It seems that the ViTVQ only consider the origion VQVAE(Not considering codes has 3 dimensions, e.g. HxWxN, N means residual times in RQVAE) https://github.com/thuanz123/enhancing-transformers/blob/1778fc497ea11ed2cef134404f99d4d6b921cda9/enhancing/modules/stage1/vitvqgan.py#L81

RQ-VAE is only mentioned here(and flag use_residual is always False) https://github.com/thuanz123/enhancing-transformers/blob/1778fc497ea11ed2cef134404f99d4d6b921cda9/enhancing/modules/stage1/quantizers.py#L42

Did I miss something? Looking forward to your reply!

JJJYmmm avatar Apr 10 '24 06:04 JJJYmmm