fixed default labels for QA model (#8399)

61073099 · Manav Rathod · GitHub · 0b02489b · 61073099
Unverified Commit 61073099 authored Nov 08, 2020 by Manav Rathod Committed by GitHub Nov 08, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/trainer.py src/transformers/trainer.py +1 -1

No files found.
--- 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"]
        )