retribert.rst 1.5 KB
Newer Older
Yacine Jernite's avatar
Yacine Jernite committed
1
RetriBERT
Sylvain Gugger's avatar
Sylvain Gugger committed
2
-----------------------------------------------------------------------------------------------------------------------
Yacine Jernite's avatar
Yacine Jernite committed
3
4

Overview
Sylvain Gugger's avatar
Sylvain Gugger committed
5
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Yacine Jernite's avatar
Yacine Jernite committed
6

Sylvain Gugger's avatar
Sylvain Gugger committed
7
8
9
The RetriBERT model was proposed in the blog post `Explain Anything Like I'm Five: A Model for Open Domain Long Form
Question Answering <https://yjernite.github.io/lfqa.html>`__. RetriBERT is a small model that uses either a single or
pair of BERT encoders with lower-dimension projection for dense semantic indexing of text.
Yacine Jernite's avatar
Yacine Jernite committed
10

Sylvain Gugger's avatar
Sylvain Gugger committed
11
12
Code to train and use the model can be found `here
<https://github.com/huggingface/transformers/tree/master/examples/distillation>`__.
Yacine Jernite's avatar
Yacine Jernite committed
13
14
15


RetriBertConfig
Sylvain Gugger's avatar
Sylvain Gugger committed
16
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Yacine Jernite's avatar
Yacine Jernite committed
17
18
19
20
21
22

.. autoclass:: transformers.RetriBertConfig
    :members:


RetriBertTokenizer
Sylvain Gugger's avatar
Sylvain Gugger committed
23
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Yacine Jernite's avatar
Yacine Jernite committed
24
25
26
27
28
29

.. autoclass:: transformers.RetriBertTokenizer
    :members:


RetriBertTokenizerFast
Sylvain Gugger's avatar
Sylvain Gugger committed
30
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Yacine Jernite's avatar
Yacine Jernite committed
31
32
33
34
35
36

.. autoclass:: transformers.RetriBertTokenizerFast
    :members:


RetriBertModel
Sylvain Gugger's avatar
Sylvain Gugger committed
37
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Yacine Jernite's avatar
Yacine Jernite committed
38
39

.. autoclass:: transformers.RetriBertModel
Sylvain Gugger's avatar
Sylvain Gugger committed
40
    :members: forward