deepmind-research/scratchgan

ScratchGAN

This is the example code for the following NeurIPS 2019 paper. If you use the code here please cite this paper:

@article{DBLP:journals/corr/abs-1905-09922,
  author    = {Cyprien de Masson d'Autume and
               Mihaela Rosca and
               Jack W. Rae and
               Shakir Mohamed},
  title     = {Training language GANs from Scratch},
  journal   = {CoRR},
  volume    = {abs/1905.09922},
  year      = {2019},
  url       = {http://arxiv.org/abs/1905.09922},
  archivePrefix = {arXiv},
  eprint    = {1905.09922},
  timestamp = {Wed, 29 May 2019 11:27:50 +0200},
  biburl    = {https://dblp.org/rec/bib/journals/corr/abs-1905-09922},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}

Contents

The code contains:

• generators.py: implementation of the generator.
• discriminator_nets.py: implementation of the discriminator.
• eval_metrics.py: implementation of the FED metric.
• losses.py: implementation of the RL loss for the generator.
• reader.py: data reader / tokenizer.
• experiment.py: main training script.

The data contains:

• {train,valid,test}.json: the EMNLP2017 News dataset.
• glove_emnlp2017.txt: the relevant subset of GloVe embeddings.

Running

./scratchgan/run.sh