lucidrains/lightweight-gan
PyTorch implementation of a lightweight GAN model for training high-resolution image generators on a single GPU within days.

Velocity · 7d
+0.8
★ / day
Trend
→steady
star history
This repository implements the ’lightweight’ GAN architecture proposed in ICLR 2021. The model uses skip-layer excitation in the generator combined with autoencoding self-supervised learning in the discriminator to achieve high-quality 1024x1024 image generation with reduced computational requirements. It trains efficiently on a single GPU and includes configurable augmentation strategies for low-data settings.