From adb8c93134f02fd0eac2b52189364af21977004c Mon Sep 17 00:00:00 2001 From: Jared Nielsen Date: Tue, 28 Jan 2020 17:24:42 -0800 Subject: [PATCH] Remove lines causing a KeyError --- examples/run_tf_glue.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/examples/run_tf_glue.py b/examples/run_tf_glue.py index 8398ccb4cb..dae11d22b3 100644 --- a/examples/run_tf_glue.py +++ b/examples/run_tf_glue.py @@ -99,9 +99,6 @@ if TASK == "mrpc": inputs_1 = tokenizer.encode_plus(sentence_0, sentence_1, add_special_tokens=True, return_tensors="pt") inputs_2 = tokenizer.encode_plus(sentence_0, sentence_2, add_special_tokens=True, return_tensors="pt") - del inputs_1["special_tokens_mask"] - del inputs_2["special_tokens_mask"] - pred_1 = pytorch_model(**inputs_1)[0].argmax().item() pred_2 = pytorch_model(**inputs_2)[0].argmax().item() print("sentence_1 is", "a paraphrase" if pred_1 else "not a paraphrase", "of sentence_0")