Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "e008d520bb9df64e7c2f644fd558301fd4276d51"
Commit 7e957237 authored by LysandreJik's avatar LysandreJik
Browse files

[Doc] XLM + Torch in documentation

parent 302a4813
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 4 deletions
docs/requirements.txt
View file @ 7e957237
......@@ -28,4 +28,5 @@ sphinxcontrib-serializinghtml==1.1.3
urllib3==1.25.3
sphinx-markdown-tables==0.0.9
numpy==1.17.2
tensorflow==2.0.0rc2
\ No newline at end of file
tensorflow==2.0.0rc2
torch==1.1.0
\ No newline at end of file
docs/source/model_doc/xlm.rst
View file @ 7e957237
......@@ -51,19 +51,19 @@ XLM
``TFXLMWithLMHeadModel``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: pytorch_transformers.TFXLMWithLMHeadModel
.. autoclass:: transformers.TFXLMWithLMHeadModel
:members:
``TFXLMForSequenceClassification``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: pytorch_transformers.TFXLMForSequenceClassification
.. autoclass:: transformers.TFXLMForSequenceClassification
:members:
``TFXLMForQuestionAnsweringSimple``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: pytorch_transformers.TFXLMForQuestionAnsweringSimple
.. autoclass:: transformers.TFXLMForQuestionAnsweringSimple
:members:
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment