gpt2.rst 940 Bytes
Newer Older
1
2
3
OpenAI GPT2
----------------------------------------------------

4
``GPT2Config``
5
6
~~~~~~~~~~~~~~~~~~~~~

7
.. autoclass:: transformers.GPT2Config
8
    :members:
9
10


11
12
``GPT2Tokenizer``
~~~~~~~~~~~~~~~~~~~~~
13

14
.. autoclass:: transformers.GPT2Tokenizer
15
    :members:
16
17


18
``GPT2Model``
19
20
~~~~~~~~~~~~~~~~~~~~~

21
.. autoclass:: transformers.GPT2Model
22
23
24
    :members:


25
``GPT2LMHeadModel``
26
27
~~~~~~~~~~~~~~~~~~~~~~~~~~~

28
.. autoclass:: transformers.GPT2LMHeadModel
29
30
31
    :members:


32
``GPT2DoubleHeadsModel``
33
34
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

35
.. autoclass:: transformers.GPT2DoubleHeadsModel
36
    :members:
LysandreJik's avatar
LysandreJik committed
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57


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

.. autoclass:: pytorch_transformers.TFGPT2Model
    :members:


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

.. autoclass:: pytorch_transformers.TFGPT2LMHeadModel
    :members:


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

.. autoclass:: pytorch_transformers.TFGPT2DoubleHeadsModel
    :members: