Skip to content

Files

scratchgan

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Nov 11, 2020
Nov 21, 2019
Jan 15, 2020
Jan 15, 2020
May 26, 2022
May 26, 2022
May 26, 2022
May 26, 2022
Sep 18, 2020
Nov 11, 2020
May 26, 2022

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