auto.rst 4.26 KB
Newer Older
1
AutoClasses
Sylvain Gugger's avatar
Sylvain Gugger committed
2
-----------------------------------------------------------------------------------------------------------------------
thomwolf's avatar
thomwolf committed
3

4
In many cases, the architecture you want to use can be guessed from the name or the path of the pretrained model you
5
are supplying to the :obj:`from_pretrained()` method.
6
AutoClasses are here to do this job for you so that you automatically retrieve the relevant model given the name/path
7
to the pretrained weights/config/vocabulary.
thomwolf's avatar
thomwolf committed
8

9
10
Instantiating one of :class:`~transformers.AutoConfig`, :class:`~transformers.AutoModel`, and
:class:`~transformers.AutoTokenizer` will directly create a class of the relevant architecture. For instance
thomwolf's avatar
thomwolf committed
11
12


13
14
15
16
.. code-block:: python

    model = AutoModel.from_pretrained('bert-base-cased')

sgugger's avatar
Extra )  
sgugger committed
17
will create a model that is an instance of :class:`~transformers.BertModel`.
18
19
20
21
22

There is one class of :obj:`AutoModel` for each task, and for each backend (PyTorch or TensorFlow).


AutoConfig
Sylvain Gugger's avatar
Sylvain Gugger committed
23
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
thomwolf's avatar
thomwolf committed
24

25
.. autoclass:: transformers.AutoConfig
thomwolf's avatar
thomwolf committed
26
27
28
    :members:


29
AutoTokenizer
Sylvain Gugger's avatar
Sylvain Gugger committed
30
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
31
32
33
34
35

.. autoclass:: transformers.AutoTokenizer
    :members:


36
AutoModel
Sylvain Gugger's avatar
Sylvain Gugger committed
37
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
thomwolf's avatar
thomwolf committed
38

39
.. autoclass:: transformers.AutoModel
thomwolf's avatar
thomwolf committed
40
41
42
    :members:


43
AutoModelForPreTraining
Sylvain Gugger's avatar
Sylvain Gugger committed
44
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
thomwolf's avatar
thomwolf committed
45
46
47
48
49

.. autoclass:: transformers.AutoModelForPreTraining
    :members:


50
AutoModelWithLMHead
Sylvain Gugger's avatar
Sylvain Gugger committed
51
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
thomwolf's avatar
thomwolf committed
52

Lysandre's avatar
Lysandre committed
53
.. autoclass:: transformers.AutoModelWithLMHead
thomwolf's avatar
thomwolf committed
54
    :members:
Lysandre's avatar
Lysandre committed
55
56


57
AutoModelForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
58
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
59
60
61
62
63

.. autoclass:: transformers.AutoModelForSequenceClassification
    :members:


64
AutoModelForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
65
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
66

67
.. autoclass:: transformers.AutoModelForMultipleChoice
Lysandre's avatar
Lysandre committed
68
69
70
    :members:


71
AutoModelForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
72
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Lysandre's avatar
Lysandre committed
73
74
75
76

.. autoclass:: transformers.AutoModelForTokenClassification
    :members:

77
78

AutoModelForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
79
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
80
81
82
83
84
85

.. autoclass:: transformers.AutoModelForQuestionAnswering
    :members:


TFAutoModel
Sylvain Gugger's avatar
Sylvain Gugger committed
86
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
87
88
89
90
91

.. autoclass:: transformers.TFAutoModel
    :members:


92
TFAutoModelForPreTraining
Sylvain Gugger's avatar
Sylvain Gugger committed
93
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
94
95
96
97
98

.. autoclass:: transformers.TFAutoModelForPreTraining
    :members:


99
TFAutoModelWithLMHead
Sylvain Gugger's avatar
Sylvain Gugger committed
100
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
101
102
103
104
105

.. autoclass:: transformers.TFAutoModelWithLMHead
    :members:


106
TFAutoModelForSequenceClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
107
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
108
109
110
111
112

.. autoclass:: transformers.TFAutoModelForSequenceClassification
    :members:


113
TFAutoModelForMultipleChoice
Sylvain Gugger's avatar
Sylvain Gugger committed
114
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
115

116
.. autoclass:: transformers.TFAutoModelForMultipleChoice
117
118
119
    :members:


120
TFAutoModelForTokenClassification
Sylvain Gugger's avatar
Sylvain Gugger committed
121
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
122
123
124

.. autoclass:: transformers.TFAutoModelForTokenClassification
    :members:
125
126
127


TFAutoModelForQuestionAnswering
Sylvain Gugger's avatar
Sylvain Gugger committed
128
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
129
130
131

.. autoclass:: transformers.TFAutoModelForQuestionAnswering
    :members: