diff --git a/docs/source/main_classes/model.rst b/docs/source/main_classes/model.rst index d22467f907..179ec53d0e 100644 --- a/docs/source/main_classes/model.rst +++ b/docs/source/main_classes/model.rst @@ -13,3 +13,9 @@ The base class ``PreTrainedModel`` implements the common methods for loading/sav .. autoclass:: transformers.PreTrainedModel :members: + +``TFPreTrainedModel`` +~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFPreTrainedModel + :members: diff --git a/docs/source/model_doc/bert.rst b/docs/source/model_doc/bert.rst index 9f2caf3e80..cfc221d381 100644 --- a/docs/source/model_doc/bert.rst +++ b/docs/source/model_doc/bert.rst @@ -70,3 +70,59 @@ BERT .. autoclass:: transformers.BertForQuestionAnswering :members: + +``TFBertModel`` +~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFBertModel + :members: + + +``TFBertForPreTraining`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFBertForPreTraining + :members: + + +``TFBertForMaskedLM`` +~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFBertForMaskedLM + :members: + + +``TFBertForNextSentencePrediction`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFBertForNextSentencePrediction + :members: + + +``TFBertForSequenceClassification`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFBertForSequenceClassification + :members: + + +``TFBertForMultipleChoice`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFBertForMultipleChoice + :members: + + +``TFBertForTokenClassification`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFBertForTokenClassification + :members: + + +``TFBertForQuestionAnswering`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFBertForQuestionAnswering + :members: + diff --git a/docs/source/model_doc/distilbert.rst b/docs/source/model_doc/distilbert.rst index de1ac73675..6d0fdcd4f3 100644 --- a/docs/source/model_doc/distilbert.rst +++ b/docs/source/model_doc/distilbert.rst @@ -41,3 +41,30 @@ DistilBERT .. autoclass:: transformers.DistilBertForQuestionAnswering :members: + +``TFDistilBertModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFDistilBertModel + :members: + + +``TFDistilBertForMaskedLM`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFDistilBertForMaskedLM + :members: + + +``TFDistilBertForSequenceClassification`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFDistilBertForSequenceClassification + :members: + + +``TFDistilBertForQuestionAnswering`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFDistilBertForQuestionAnswering + :members: diff --git a/docs/source/model_doc/gpt.rst b/docs/source/model_doc/gpt.rst index 39995a98fc..dd6e314b8f 100644 --- a/docs/source/model_doc/gpt.rst +++ b/docs/source/model_doc/gpt.rst @@ -34,3 +34,24 @@ OpenAI GPT .. autoclass:: transformers.OpenAIGPTDoubleHeadsModel :members: + + +``TFOpenAIGPTModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFOpenAIGPTModel + :members: + + +``TFOpenAIGPTLMHeadModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFOpenAIGPTLMHeadModel + :members: + + +``TFOpenAIGPTDoubleHeadsModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFOpenAIGPTDoubleHeadsModel + :members: diff --git a/docs/source/model_doc/gpt2.rst b/docs/source/model_doc/gpt2.rst index 92decb14de..19b435fc03 100644 --- a/docs/source/model_doc/gpt2.rst +++ b/docs/source/model_doc/gpt2.rst @@ -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: diff --git a/docs/source/model_doc/roberta.rst b/docs/source/model_doc/roberta.rst index 5351e018cd..ed839456af 100644 --- a/docs/source/model_doc/roberta.rst +++ b/docs/source/model_doc/roberta.rst @@ -34,3 +34,24 @@ RoBERTa .. autoclass:: transformers.RobertaForSequenceClassification :members: + + +``TFRobertaModel`` +~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFRobertaModel + :members: + + +``TFRobertaForMaskedLM`` +~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFRobertaForMaskedLM + :members: + + +``TFRobertaForSequenceClassification`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFRobertaForSequenceClassification + :members: diff --git a/docs/source/model_doc/transformerxl.rst b/docs/source/model_doc/transformerxl.rst index c8a9cc7d99..243726aa95 100644 --- a/docs/source/model_doc/transformerxl.rst +++ b/docs/source/model_doc/transformerxl.rst @@ -28,3 +28,17 @@ Transformer XL .. autoclass:: transformers.TransfoXLLMHeadModel :members: + + +``TFTransfoXLModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFTransfoXLModel + :members: + + +``TFTransfoXLLMHeadModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFTransfoXLLMHeadModel + :members: diff --git a/docs/source/model_doc/xlm.rst b/docs/source/model_doc/xlm.rst index 344371ad52..367938fab5 100644 --- a/docs/source/model_doc/xlm.rst +++ b/docs/source/model_doc/xlm.rst @@ -39,3 +39,31 @@ XLM .. autoclass:: transformers.XLMForQuestionAnswering :members: + + +``TFXLMModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFXLMModel + :members: + + +``TFXLMWithLMHeadModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFXLMWithLMHeadModel + :members: + + +``TFXLMForSequenceClassification`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFXLMForSequenceClassification + :members: + + +``TFXLMForQuestionAnswering`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFXLMForQuestionAnswering + :members: diff --git a/docs/source/model_doc/xlnet.rst b/docs/source/model_doc/xlnet.rst index a471506278..000f163ec0 100644 --- a/docs/source/model_doc/xlnet.rst +++ b/docs/source/model_doc/xlnet.rst @@ -41,3 +41,31 @@ XLNet .. autoclass:: transformers.XLNetForQuestionAnswering :members: + + +``TFXLNetModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFXLNetModel + :members: + + +``TFXLNetLMHeadModel`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFXLNetLMHeadModel + :members: + + +``TFXLNetForSequenceClassification`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFXLNetForSequenceClassification + :members: + + +``TFXLNetForQuestionAnswering`` +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: pytorch_transformers.TFXLNetForQuestionAnswering + :members: