This commit is contained in:
Soumith Chintala 2017-01-30 19:59:17 +05:30
parent 140f5bfb0e
commit 64d35231af
1 changed files with 27 additions and 0 deletions

27
README.md Normal file
View File

@ -0,0 +1,27 @@
Wasserstein GAN
===============
Code accompanying the paper ["Wasserstein GAN"](https://arxiv.org/abs/1701.07875)
##Prerequisites
- Computer with Linux or OSX
- [PyTorch](http://pytorch.org)
- For training, an NVIDIA GPU is strongly recommended for speed. CPU is supported but training is very slow.
##Reproducing LSUN experiments
**With DCGAN:**
```python
python main.py --dataset lsun --dataroot [lsun-train-folder] --cuda
```
**With MLP:**
```python
python main.py --mlp_G --ngf 512
```
More improved README in the works.