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}
}
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.
./scratchgan/run.sh