Generative Adversarial Transformers (Pytorch / TF)

Official Implementation of the Generative Adversarial Transformers paper, in both pytorch and tensorflow, for image and compositional scene generation. The codebase supports training, evaluation, image sampling, and variety of visualizations.

Updates for version 1.5.2 (Feb 22, 2022): We updated the weight initialization of the PyTorch version to the intended scale, leading to a substantial improvement in the model's learning speed.

Generative Adversarial Transformers

Official Implementation of the Generative Adversarial Transformers paper for image and compositional scene generation.
The codebase supports training, evaluation, image sampling, and variety of visualizations.