fixed default labels for QA model (#8399)

2020-11-08 09:08:14 -05:00 · 2020-11-08 09:08:14 -05:00 · 610730998f
parent 0b02489b2c
commit 610730998f
1 changed files with 1 additions and 1 deletions
--- 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"]
        )