← all repositories

ajbrock/BigGAN-PyTorch

A PyTorch implementation of BigGAN for training large-scale generative adversarial networks on high-resolution image synthesis.

2.9k stars Python Image · Video · Audio
BigGAN-PyTorch
Velocity · 7d
+1.1
★ / day
Trend
steady
star history

This repository provides an unofficial PyTorch implementation of BigGAN based on the paper Large Scale GAN Training for High Fidelity Natural Image Synthesis. It enables training of BigGAN models on ImageNet with support for 4-8 GPU configurations using gradient accumulation to achieve effective mega-batch sizes. The code handles dataset preparation, inception score calculation, and FID evaluation, with pretrained models available for download.

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