SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

TF models added to documentation

Browse Source
This commit is contained in:
LysandreJik
2019-09-25 06:31:05 -04:00
committed by Lysandre Debut
parent cf5c5c9e1c
commit 4acd87ff4e
9 changed files with 222 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
docs/source/model_doc/gpt2.rst
View File

@@ -34,3 +34,24 @@ OpenAI GPT2
.. autoclass:: transformers.GPT2DoubleHeadsModel
:members:
``TFGPT2Model``
~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: pytorch_transformers.TFGPT2Model
:members:
``TFGPT2LMHeadModel``
~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: pytorch_transformers.TFGPT2LMHeadModel
:members:
``TFGPT2DoubleHeadsModel``
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: pytorch_transformers.TFGPT2DoubleHeadsModel
:members: