drewszurko / tensorflow-WGAN-GP

TensorFlow 2.0 implementation of Improved Training of Wasserstein GANs

Home Page:https://arxiv.org/abs/1704.00028

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

TensorFlow 2.0 implementation of Improved Training of Wasserstein GANs

https://arxiv.org/abs/1704.00028

License:MIT License


Languages

Language:Python 100.0%