Skip to content

Latest commit

 

History

History
19 lines (11 loc) · 550 Bytes

README.md

File metadata and controls

19 lines (11 loc) · 550 Bytes

Mostly adapted from here.

Training:

python train.py

This will do a grid search on a fraction of the PTB dataset and store checkpoints of the model in a file (default is ./model.pt).

Generation:

python generate.py

The default setting is that during training in train.py a checkpoint model is written to ./model.pt by default. This model checkpoint is also used for generation in generate.py. The generated text is written to generated.txt by default.