First commit

This commit is contained in:
Jeff Wu
2019-02-10 20:22:00 -08:00
committed by Jeff Wu
commit c2dae27c10
7 changed files with 463 additions and 0 deletions

29
README.md Normal file
View File

@@ -0,0 +1,29 @@
# 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
```