Use labels to remove deprecation warnings (#4807)

2020-06-05 16:41:46 -04:00
parent 5c0cfc2cf0
commit f1fe18465d
10 changed files with 17 additions and 17 deletions
--- a/tests/test_modeling_gpt2.py
+++ b/tests/test_modeling_gpt2.py
@@ -268,7 +268,7 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase):
                "mc_token_ids": mc_token_ids,
                "attention_mask": multiple_choice_input_mask,
                "token_type_ids": multiple_choice_token_type_ids,
-                "lm_labels": multiple_choice_inputs_ids,
+                "labels": multiple_choice_inputs_ids,
            }

            loss, lm_logits, mc_logits, _ = model(**inputs)