c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Doc] XLM + Torch in documentation

Browse Source
This commit is contained in:
LysandreJik committed 2019-09-26 10:08:56 -04:00
1 parent 302a4813a5
commit 7e957237e4
2 files changed
+5 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

docs/requirements.txt
+2 -1
View File
@@ -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
tensorflow==2.0.0rc2
torch==1.1.0
docs/source/model_doc/xlm.rst
+3 -3
View File
@@ -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: