Overview

From scratch implementation of WGANs

Technical Details

• Framework: PyTorch
• Dataset: MNIST
• Category: Generative Models

Implementation Details

WGAN and WGAN-GP architecture in Pytorch

I implemented the WGAN and WGAN-GP using Pytorch on the flickr8000 dataset.

Wasserstein GAN Improved training with WGAN

Datasets

MNIST: Link

Frameworks:

Pytorch

Source Code

📁 GitHub Repository: WGANs

View the complete implementation, training scripts, and documentation on GitHub.