Skip to content

wilile26811249/ViTGAN

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits

.gitignore

.gitignore

 
 

LICENSE

LICENSE

 
 

README.md

README.md

 
 

image_dataset.py

image_dataset.py

 
 

main.py

main.py

 
 

model_ach.jpg

model_ach.jpg

 
 

models.py

models.py

 
 

utils.py

utils.py

 
 

Repository files navigation

ViTGAN: Training GANs with Vision Transformers

A PyTorch implementation of ViTGAN based on paper ViTGAN: Training GANs with Vision Transformers.

Network Architecture image from the paper

Reference

Kwonjoon Lee, Huiwen Chang, Lu Jiang, Han Zhang, Zhuowen Tu, Ce Liu. ViTGAN: Training GANs with Vision Transformers. arXiv:2107.04589v1, 2021.

Author

Hong-Jia Chen

About

A PyTorch implementation of ViTGAN based on paper ViTGAN: Training GANs with Vision Transformers.

Topics

pytorch generative-adversarial-network self-attention

Resources

Readme

License

MIT license
Activity

Stars

162 stars

Watchers

5 watching

Forks

28 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages