gpt-2/README.md

30 lines
580 B
Markdown
Raw Normal View History

2019-02-10 20:22:00 -08:00
# gpt-2
Code and samples from the paper "Language Models are Unsupervised Multitask Learners"
## Installation
Download the model data:
```
gsutil rsync -r gs://gpt-2/models/ models/
```
Install python packages:
```
pip install -r requirements.txt
```
## Sample generation
| WARNING: Samples are unfiltered and may contain offensive content. |
| --- |
To generate unconditional samples from the small model:
```
python3 src/main.py | tee samples
```
There are various flags for controlling the samples:
```
python3 src/main.py --top_k 40 --temperature 0.7 | tee samples
```