← all repositories

lucidrains/lightweight-gan

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

lightweight-gan
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.