From 95aa8290f4260c6c40177fe2801b35b709817d00 Mon Sep 17 00:00:00 2001 From: flozi00 Date: Sun, 14 Jun 2020 17:13:07 +0200 Subject: [PATCH] typo --- .../question_answering/question_answering_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/simpletransformers/question_answering/question_answering_model.py b/simpletransformers/question_answering/question_answering_model.py index 3b4c8c12..0d3a517a 100755 --- a/simpletransformers/question_answering/question_answering_model.py +++ b/simpletransformers/question_answering/question_answering_model.py @@ -704,7 +704,7 @@ def evaluate(self, eval_data, output_dir, verbose_logging=False): "token_type_ids": batch[2], } - if self.args["model_type"] in ["xlm", "roberta", "distilbert", "camembert", "electra", "xlmroberta","bart]: + if self.args["model_type"] in ["xlm", "roberta", "distilbert", "camembert", "electra", "xlmroberta","bart"]: del inputs["token_type_ids"] example_indices = batch[3]