diff --git a/README.md b/README.md new file mode 100644 index 0000000..41e1a5a --- /dev/null +++ b/README.md @@ -0,0 +1,21 @@ +--- +language: en +tags: +- exbert + +license: apache-2.0 +datasets: +- openwebtext +--- + +# DistilGPT2 + +DistilGPT2 English language model pretrained with the supervision of [GPT2](https://huggingface.co/gpt2) (the smallest version of GPT2) on [OpenWebTextCorpus](https://skylion007.github.io/OpenWebTextCorpus/), a reproduction of OpenAI's WebText dataset. The model has 6 layers, 768 dimension and 12 heads, totalizing 82M parameters (compared to 124M parameters for GPT2). On average, DistilGPT2 is two times faster than GPT2. + +On the [WikiText-103](https://blog.einstein.ai/the-wikitext-long-term-dependency-language-modeling-dataset/) benchmark, GPT2 reaches a perplexity on the test set of 16.3 compared to 21.1 for DistilGPT2 (after fine-tuning on the train set). + +We encourage to check [GPT2](https://huggingface.co/gpt2) to know more about usage, limitations and potential biases. + + + +