Shubhamai / 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.
15Updated last month

Related projects: