ElectraForQuestionAnswering (#4913)

* ElectraForQuestionAnswering

* udate __init__

* add test for electra qa model

* add ElectraForQuestionAnswering in auto models

* add ElectraForQuestionAnswering in all_model_classes

* fix outputs, input_ids defaults to None

* add ElectraForQuestionAnswering in docs

* remove commented line
This commit is contained in:
Suraj Patil
2020-06-11 00:47:52 +05:30
committed by GitHub
parent 5d63ca6c38
commit ef2dcdccaa
5 changed files with 162 additions and 0 deletions

View File

@@ -106,6 +106,13 @@ ElectraForTokenClassification
:members:
ElectraForQuestionAnswering
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.ElectraForQuestionAnswering
:members:
TFElectraModel
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~