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

Add GPT2ForSequenceClassification based on DialogRPT (#7501)

Browse Source 
* Add GPT2ForSequenceClassification based on DialogRPT

* Better documentation

* Code quality
This commit is contained in:
Lysandre Debut
2020-10-06 23:31:21 +02:00
committed by GitHub
parent 500be01c5d
commit 5982431814
5 changed files with 188 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -88,6 +88,13 @@ GPT2DoubleHeadsModel
:members: forward
GPT2ForSequenceClassification
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.GPT2ForSequenceClassification
:members: forward
TFGPT2Model
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~