Is Each VQGAN model of class TrainVQGAN and class VQmodel different?

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

Is Each VQGAN model of class TrainVQGAN and class VQmodel different? #13

Open 9B8DY6 opened 1 year ago

9B8DY6 commented 1 year ago

I am going through your MaskGIT code to study how to implement it, Thank you! But I have a question about VQGAN for tokenization. I think VQGAN for tokenization and VQGAN in training_vqgan.py are different to each other because the parameters of those are not same with each other. If I mistake it, let me know, please. Thx!