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

[Reformer] Add Masked LM Reformer (#5426)

Browse Source 
* fix conflicts

* fix

* happy rebasing
This commit is contained in:
Patrick von Platen
2020-07-01 22:43:18 +02:00
committed by GitHub
parent f4323dbf8c
commit d16e36c7e5
5 changed files with 130 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docs/source/model_doc/reformer.rst
View File

@@ -114,9 +114,15 @@ ReformerModelWithLMHead
:members:
ReformerForMaskedLM
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.ReformerForMaskedLM
:members:
ReformerForQuestionAnswering
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.ReformerForQuestionAnswering
:members: