gpt2.rst 2.54 KB
Newer Older
1
2
3
OpenAI GPT2
----------------------------------------------------

4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
Overview
~~~~~~~~~~~~~~~~~~~~~

OpenAI GPT-2 model was proposed in
`Language Models are Unsupervised Multitask Learners`_
by Alec Radford*, Jeffrey Wu*, Rewon Child, David Luan, Dario Amodei** and Ilya Sutskever**.
It's a causal (unidirectional) transformer pre-trained using  language modeling on a very large
corpus of ~40 GB of text data.

The abstract from the paper is the following:

*GPT-2 is a large transformer-based language model with 1.5 billion parameters, trained on a dataset[1]
of 8 million web pages. GPT-2 is trained with a simple objective: predict the next word, given all of the previous
words within some text. The diversity of the dataset causes this simple goal to contain naturally occurring
demonstrations of many tasks across diverse domains. GPT-2 is a direct scale-up of GPT, with more than 10X
the parameters and trained on more than 10X the amount of data.*

Tips:

- GPT-2 is a model with absolute position embeddings so it's usually advised to pad the inputs on
  the right rather than the left.
- GPT-2 was trained with a causal language modeling (CLM) objective and is therefore powerful at predicting the next
  token in a sequence. Leveraging this feature allows GPT-2 to generate syntactically coherent text as
  it can be observed in the `run_generation.py` example script.
- The PyTorch models can take the `past` as input, which is the previously computed key/value attention pairs. Using
  this `past` value prevents the model from re-computing pre-computed values in the context of text generation.
  See `reusing the past in generative models <../quickstart.html#using-the-past>`_ for more information on the usage
  of this argument.


34
``GPT2Config``
35
36
~~~~~~~~~~~~~~~~~~~~~

37
.. autoclass:: transformers.GPT2Config
38
    :members:
39
40


41
42
``GPT2Tokenizer``
~~~~~~~~~~~~~~~~~~~~~
43

44
.. autoclass:: transformers.GPT2Tokenizer
45
    :members:
46
47


48
``GPT2Model``
49
50
~~~~~~~~~~~~~~~~~~~~~

51
.. autoclass:: transformers.GPT2Model
52
53
54
    :members:


55
``GPT2LMHeadModel``
56
57
~~~~~~~~~~~~~~~~~~~~~~~~~~~

58
.. autoclass:: transformers.GPT2LMHeadModel
59
60
61
    :members:


62
``GPT2DoubleHeadsModel``
63
64
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

65
.. autoclass:: transformers.GPT2DoubleHeadsModel
66
    :members:
LysandreJik's avatar
LysandreJik committed
67
68
69
70
71


``TFGPT2Model``
~~~~~~~~~~~~~~~~~~~~~

72
.. autoclass:: transformers.TFGPT2Model
LysandreJik's avatar
LysandreJik committed
73
74
75
76
77
78
    :members:


``TFGPT2LMHeadModel``
~~~~~~~~~~~~~~~~~~~~~~~~~~~

79
.. autoclass:: transformers.TFGPT2LMHeadModel
LysandreJik's avatar
LysandreJik committed
80
81
82
83
84
85
    :members:


``TFGPT2DoubleHeadsModel``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

86
.. autoclass:: transformers.TFGPT2DoubleHeadsModel
LysandreJik's avatar
LysandreJik committed
87
    :members: