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/utils/check_repo.py
+++ b/utils/check_repo.py
@@ -29,6 +29,7 @@ TEST_FILES_WITH_NO_COMMON_TESTS = [
    "test_modeling_tf_camembert.py",
    "test_modeling_tf_xlm_roberta.py",
    "test_modeling_xlm_roberta.py",
+    "test_modeling_pegasus.py",
 ]

 # Update this list for models that are not documented with a comment explaining the reason it should not be.