Skip to content

Commit

Permalink
Add step 1 instructions for MaskGAN. (tensorflow#6917)
Browse files Browse the repository at this point in the history 
* Add step 1 instructions for MaskGAN.
  • Loading branch information
@a-dai
a-dai authored May 31, 2019
1 parent 7546a9e commit de10c0c
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions research/maskgan/README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,10 @@ tested. Pretraining may not work correctly.

For training on PTB:

1. Pretrain a LM on PTB and store the checkpoint in `/tmp/pretrain-lm/`.
Instructions WIP.
1. Follow instructions here ([Tensorflow RNN Language Model Tutorial](https://www.tensorflow.org/tutorials/sequences/recurrent)) to train a language model on PTB dataset.
Copy PTB data downloaded from the above tensorflow RNN tutorial to folder "/tmp/ptb". It should contain following three files: ptb.train.txt, ptb.test.txt, ptb.valid.txt
Make folder /tmp/pretrain-lm and copy checkpoints from above Tensorflow RNN tutorial under this folder.


2. Run MaskGAN in MLE pretraining mode. If step 1 was not run, set
`language_model_ckpt_dir` to empty.
Expand Down

0 comments on commit de10c0c

Please sign in to comment.