Move prediction_loss_only to TrainingArguments (#6426)

2020-08-12 08:03:45 -04:00
parent e9c3031463
commit 34fabe1697
5 changed files with 37 additions and 8 deletions
--- a/docs/source/model_doc/pegasus.rst
+++ b/docs/source/model_doc/pegasus.rst
@@ -66,6 +66,10 @@ PegasusForConditionalGeneration
 This class inherits all functionality from ``BartForConditionalGeneration``, see that page for method signatures.
 Available models are listed at `Model List <https://huggingface.co/models?search=pegasus>`__

+.. autoclass:: transformers.PegasusForConditionalGeneration
+    :members: generate, forward
+
+
 PegasusConfig
 ~~~~~~~~~~~~~~~~~~~
 This config fully inherits from ``BartConfig``, but pegasus uses different default values: