Skip to content

GitLab

Unverified Commit a2096917 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Add AlbertForPretraining to doc (#5914)

parent f1a4e06f
Hide whitespace changes
Inline Side-by-side
Showing with 14 additions and 0 deletions
docs/source/model_doc/albert.rst
View file @ a2096917
...@@ -61,6 +61,13 @@ AlbertModel ...@@ -61,6 +61,13 @@ AlbertModel
:members: :members:
AlbertForPreTraining
~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.AlbertForPreTraining
:members:
AlbertForMaskedLM AlbertForMaskedLM
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
...@@ -103,6 +110,13 @@ TFAlbertModel ...@@ -103,6 +110,13 @@ TFAlbertModel
:members: :members:
TFAlbertForPreTraining
~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: transformers.TFAlbertForPreTraining
:members:
TFAlbertForMaskedLM TFAlbertForMaskedLM
~~~~~~~~~~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~~
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment