Skip to content
TensorFlow 2.0 implementation of Improved Training of Wasserstein GANs
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.

Files

Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
LICENSE
README.md
main.py
models.py
ops.py
requirements.txt
utils.py

README.md

TensorFlow 2.0 WGAN-GP

TensorFlow 2.0 implementation of Improved Training of Wasserstein GANs [1]. New/existing TensorFlow features found in this repository include eager execution, AutoGraph, Keras high-level API, and TensorFlow Datasets.

Requirements

Datasets

Usage

Install requirements

$ pip install -r requirements.txt

Train model

$ python main.py -dataset celeb_a -batch_size 64 -image_size 64

References

[1] Improved Training of Wasserstein GANs

You can’t perform that action at this time.