/pytorch-vqgan

This repo contains the implementation of VQGAN, Taming Transformers for High-Resolution Image Synthesis in PyTorch from scratch. I have added support for custom datasets, testings, experiment tracking etc.

Primary LanguagePythonMIT LicenseMIT

Stargazers