From 610730998fe542ad2fe3515e7ea79633a3086e64 Mon Sep 17 00:00:00 2001 From: Manav Rathod Date: Sun, 8 Nov 2020 09:08:14 -0500 Subject: [PATCH] fixed default labels for QA model (#8399) --- src/transformers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index 62200d5976..8166b2ed06 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -328,7 +328,7 @@ class Trainer: self.hp_search_backend = None self.use_tune_checkpoints = False default_label_names = ( - ["start_positions, end_positions"] + ["start_positions", "end_positions"] if type(self.model) in MODEL_FOR_QUESTION_ANSWERING_MAPPING.values() else ["labels"] )