roberta.rst 1.8 KB
Newer Older
LysandreJik's avatar
Doc  
LysandreJik committed
1
2
3
RoBERTa
----------------------------------------------------

Lysandre's avatar
Lysandre committed
4
5
6
7
8
9
10
11
12
13
14
The RoBERTa model was proposed in `RoBERTa: A Robustly Optimized BERT Pretraining Approach`_
by Yinhan Liu, Myle Ott, Naman Goyal, Jingfei Du, Mandar Joshi, Danqi Chen, Omer Levy, Mike Lewis, Luke Zettlemoyer,
Veselin Stoyanov. It is based on Google's BERT model released in 2018.

It builds on BERT and modifies key hyperparameters, removing the next-sentence pretraining
objective and training with much larger mini-batches and learning rates.

This implementation is the same as BertModel with a tiny embeddings tweak as well as a setup for Roberta pretrained
models.

RobertaConfig
LysandreJik's avatar
Doc  
LysandreJik committed
15
16
~~~~~~~~~~~~~~~~~~~~~

17
.. autoclass:: transformers.RobertaConfig
LysandreJik's avatar
Doc  
LysandreJik committed
18
19
20
    :members:


Lysandre's avatar
Lysandre committed
21
RobertaTokenizer
LysandreJik's avatar
Doc  
LysandreJik committed
22
23
~~~~~~~~~~~~~~~~~~~~~

24
.. autoclass:: transformers.RobertaTokenizer
LysandreJik's avatar
Doc  
LysandreJik committed
25
26
27
    :members:


Lysandre's avatar
Lysandre committed
28
RobertaModel
LysandreJik's avatar
Doc  
LysandreJik committed
29
30
~~~~~~~~~~~~~~~~~~~~

31
.. autoclass:: transformers.RobertaModel
LysandreJik's avatar
Doc  
LysandreJik committed
32
33
34
    :members:


Lysandre's avatar
Lysandre committed
35
RobertaForMaskedLM
LysandreJik's avatar
Doc  
LysandreJik committed
36
37
~~~~~~~~~~~~~~~~~~~~~~~~~~

38
.. autoclass:: transformers.RobertaForMaskedLM
LysandreJik's avatar
Doc  
LysandreJik committed
39
40
41
    :members:


Lysandre's avatar
Lysandre committed
42
RobertaForSequenceClassification
LysandreJik's avatar
Doc  
LysandreJik committed
43
44
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

45
.. autoclass:: transformers.RobertaForSequenceClassification
LysandreJik's avatar
Doc  
LysandreJik committed
46
    :members:
LysandreJik's avatar
LysandreJik committed
47
48


Lysandre's avatar
Lysandre committed
49
50
51
52
53
54
RobertaForTokenClassification
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: transformers.RobertaForTokenClassification
    :members:

Lysandre's avatar
Lysandre committed
55
TFRobertaModel
LysandreJik's avatar
LysandreJik committed
56
57
~~~~~~~~~~~~~~~~~~~~

58
.. autoclass:: transformers.TFRobertaModel
LysandreJik's avatar
LysandreJik committed
59
60
61
    :members:


Lysandre's avatar
Lysandre committed
62
TFRobertaForMaskedLM
LysandreJik's avatar
LysandreJik committed
63
64
~~~~~~~~~~~~~~~~~~~~~~~~~~

65
.. autoclass:: transformers.TFRobertaForMaskedLM
LysandreJik's avatar
LysandreJik committed
66
67
68
    :members:


Lysandre's avatar
Lysandre committed
69
TFRobertaForSequenceClassification
LysandreJik's avatar
LysandreJik committed
70
71
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

72
.. autoclass:: transformers.TFRobertaForSequenceClassification
LysandreJik's avatar
LysandreJik committed
73
    :members:
Lysandre's avatar
Lysandre committed
74
75
76
77
78
79
80


TFRobertaForTokenClassification
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

.. autoclass:: transformers.TFRobertaForTokenClassification
    :members: